Move to top level

apple · Jan 8, 2025 · a9ca036 · a9ca036
1 parent 4a434c8
commit a9ca036
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/axlearn/common/flash_attention/conftest.py → conftest.py b/axlearn/common/flash_attention/conftest.py → conftest.py
@@ -2,7 +2,11 @@
 
 """Configures pytest to distribute tests to multiple GPUs.
 
-Usage on 8 GPU machines: PARALLEL_GPU_TEST=1 pytest -n 8 gpu_attention_test.py
+This is not enabled by default and requires explicit opt-in by setting the environment variable
+PARALLEL_GPU_TEST. This is because not all GPU tests are single-GPU tests.
+
+Example usage on 8 GPU machines:
+PARALLEL_GPU_TEST=1 pytest -n 8 axlearn/common/flash_attention/gpu_attention_test.py
 """
 import os