From 4159ab3cb047d704acb1e96b6c8c9c9367820283 Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Fri, 8 Nov 2024 12:23:17 -0500 Subject: [PATCH] Fix README (#96) * Update README.md * fix readme --- README.md | 2 +- flash_attn/flash_attn_triton_amd/README.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 35debfac7..f25465af1 100644 --- 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 index 798d78a12..fc411425f 100644 --- 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