File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -36,7 +36,7 @@ export XLA_FLAGS="${XLA_FLAGS} --xla_gpu_deterministic_ops"
3636python3 -m pytest -c $TE_PATH /tests/jax/pytest.ini -v --junitxml=$XML_LOG_DIR /pytest_test_single_gpu_encoder.xml $TE_PATH /examples/jax/encoder/test_single_gpu_encoder.py || test_fail " test_single_gpu_encoder.py"
3737# Test without custom calls
3838export XLA_FLAGS=" ${XLA_FLAGS} --xla_gpu_deterministic_ops"
39- NVTE_JAX_CUSTOM_CALLS=" false" python3 -m pytest -c $TE_PATH /tests/jax/pytest.ini -v --junitxml=$XML_LOG_DIR /pytest_test_single_gpu_encoder .xml $TE_PATH /examples/jax/encoder/test_single_gpu_encoder.py || test_fail " test_single_gpu_encoder.py without custom calls"
39+ NVTE_JAX_CUSTOM_CALLS=" false" python3 -m pytest -c $TE_PATH /tests/jax/pytest.ini -v --junitxml=$XML_LOG_DIR /pytest_test_single_gpu_encoder_without_custom_call .xml $TE_PATH /examples/jax/encoder/test_single_gpu_encoder.py || test_fail " test_single_gpu_encoder.py without custom calls"
4040
4141if [ $RET -ne 0 ]; then
4242 echo " Error: some sub-tests failed: $FAILED_CASES "
You can’t perform that action at this time.
0 commit comments