Code

bench & add more tests#130

Code

Draft

micmelesse wants to merge 13 commits intomain_perfROCm/flash-attention:main_perffrom micmelesse/test_allROCm/flash-attention:micmelesse/test_allCopy head branch name to clipboard

+1,256-643

Commits

Commits on Mar 5, 2025

• test and bench work compressed

  Show description for 1c0c99d
  micmelesse

  committed
  1c0c99d

  Copy full SHA for 1c0c99d

• mv test_op_prefill_bwd_split_impl

  micmelesse

  committed
  cd03470

  Copy full SHA for cd03470

• save test

  micmelesse

  committed
  d58949b

  Copy full SHA for d58949b

• test ir for sanity

  micmelesse

  committed
  b72133f

  Copy full SHA for b72133f

Commits on Mar 6, 2025

• test qkv ir

  micmelesse

  committed
  020b1ca

  Copy full SHA for 020b1ca

• use input helper

  micmelesse

  committed
  95317b9

  Copy full SHA for 95317b9

• kvpacked benching added

  micmelesse

  committed
  12d0e19

  Copy full SHA for 12d0e19

Commits on Mar 7, 2025

• output do from the lower level functions

  micmelesse

  committed
  12301e7

  Copy full SHA for 12301e7

• clean up packing input changing

  micmelesse

  committed
  58dff07

  Copy full SHA for 58dff07

• clean up bwd

  micmelesse

  committed
  9d85d61

  Copy full SHA for 9d85d61

• add qkv packed

  micmelesse

  committed
  c1a9a69

  Copy full SHA for c1a9a69

• add causal and dropout as a config

  micmelesse

  committed
  af1c895

  Copy full SHA for af1c895

• test all normal configs

  micmelesse

  committed
  dad31be

  Copy full SHA for dad31be