File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -28,7 +28,7 @@ NVTE_GROUPED_LINEAR_SINGLE_PARAM=1 python3 -m pytest --tb=auto --junitxml=$XML_L
2828python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_recipe.xml $TE_PATH /tests/pytorch/test_recipe.py || test_fail " test_recipe.py"
2929python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_custom_recipe.xml $TE_PATH /tests/pytorch/test_custom_recipe.py || test_fail " test_custom_recipe.py"
3030python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_deferred_init.xml $TE_PATH /tests/pytorch/test_deferred_init.py || test_fail " test_deferred_init.py"
31- PYTORCH_JIT=0 NVTE_TORCH_COMPILE=0 NVTE_ALLOW_NONDETERMINISTIC_ALGO=0 NVTE_FUSED_ATTN=0 NVIDIA_TF32_OVERRIDE=0 python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_numerics.xml $TE_PATH /tests/pytorch/test_numerics.py || test_fail " test_numerics.py"
31+ PYTORCH_JIT=0 NVTE_TORCH_COMPILE=0 NVTE_ALLOW_NONDETERMINISTIC_ALGO=0 NVTE_FUSED_ATTN=0 python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_numerics.xml $TE_PATH /tests/pytorch/test_numerics.py || test_fail " test_numerics.py"
3232PYTORCH_JIT=0 NVTE_TORCH_COMPILE=0 NVTE_ALLOW_NONDETERMINISTIC_ALGO=0 NVTE_FUSED_ATTN=0 python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_cuda_graphs.xml $TE_PATH /tests/pytorch/test_cuda_graphs.py || test_fail " test_cuda_graphs.py"
3333python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_jit.xml $TE_PATH /tests/pytorch/test_jit.py || test_fail " test_jit.py"
3434python3 -m pytest --tb=auto --junitxml=$XML_LOG_DIR /pytest_test_fused_rope.xml $TE_PATH /tests/pytorch/test_fused_rope.py || test_fail " test_fused_rope.py"
You can’t perform that action at this time.
0 commit comments