Fix README (#96)

* Update README.md * fix readme
ROCm · Nov 8, 2024 · 4159ab3 · 4159ab3
1 parent bf67fc9
commit 4159ab3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -166,7 +166,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
 ```
 
 

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