Merge branch 'ko3n1g/tests/fix-flash-fused-attn' into 'main'

tests: Set flash/fused attn See merge request ADLR/megatron-lm!2304
NVIDIA · Nov 6, 2024 · 0594f20 · 0594f20
2 parents f39c48d + 1bb6c94
commit 0594f20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit_tests/conftest.py b/tests/unit_tests/conftest.py
@@ -13,7 +13,7 @@
 from tests.unit_tests.test_utilities import Utils
 
 
-@pytest.fixture(scope="session", autouse=True)
+@pytest.fixture(scope="function", autouse=True)
 def set_env():
     if is_te_min_version("1.3"):
         os.environ['NVTE_FLASH_ATTN'] = '0'