updating README.md

erfanzar · Oct 23, 2024 · 2f1ca04 · 2f1ca04
1 parent dda2760
commit 2f1ca04
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -97,6 +97,7 @@ attention = get_cached_flash_attention(
 ### Environment Variables
 
 - `FORCE_MHA`: Set to "true", "1", or "on" to force using MHA implementation even for GQA cases
+- `FLASH_ATTN_BLOCK_PTR`: set to "1" to use `tl.make_block_ptr` for accessing pointer in fwd mode (better for H100/H200 GPUs)
 
 ## Performance Tips