ROCm / flash-attention

Fast and memory-efficient exact attention
BSD 3-Clause "New" or "Revised" License
131 stars 41 forks source link

Remove Hardcoded Building Options #19

Closed jayz0123 closed 10 months ago

jayz0123 commented 10 months ago

rocm5.6_ubuntu20.04_py3.8_pytorch_1.12.1 9119 passed, 32 skipped in 390.07s (0:06:30)

rocm5.6_ubuntu20.04_py3.8_pytorch_1.13.1 9119 passed, 32 skipped in 392.50s (0:06:32)

rocm5.6_ubuntu20.04_py3.10_pytorch_2.0.1 9119 passed, 32 skipped in 396.98s (0:06:36)

rocm5.7_ubuntu22.04_py3.10_pytorch_2.0.1 9119 passed, 32 skipped in 394.38s (0:06:34)

sabreshao commented 10 months ago

Please paste a typical test result and give a list here including all docker images with which you have performed testing against this PR.

fsx950223 commented 10 months ago

Add python format too, black or pylint. Are these lint configs accepted by upstream?