Update README.md

ROCm · Nov 8, 2024 · 943af97 · 943af97
1 parent bf67fc9
commit 943af97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash_attn/flash_attn_triton_amd/README.md b/flash_attn/flash_attn_triton_amd/README.md
@@ -40,7 +40,7 @@ Then install and test Flash Attention with the flag `FLASH_ATTENTION_TRITON_AMD_
 export FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
 cd flash-attention
 python setup.py install
-pytest tests/test_flash_attn.py
+pytest tests/test_flash_attn_triton_amd.py
 ```
 
 #### Credits