Flexstorydiff / xformers /tests /readme_test_on_rocm.txt
FlexTheAi's picture
Upload folder using huggingface_hub
e202b16 verified
raw
history blame contribute delete
341 Bytes
1. #> pip install -e ./
2. verify testing for generic fmha inference on ROCM
#> pytest tests/test_mem_eff_attention.py::test_forward
3. verify testing for decoder fmha inference on ROCM
#> pytest tests/test_mem_eff_attention.py::test_decoder
#> pytest tests/test_mem_eff_attention.py::test_splitk_decoder