Skip to content

Commit 4d14578

Browse files
authored
[JAX] Fix XML filename in the L0_jax_uniitest (#2205)
fix xml file name Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
1 parent 9e72796 commit 4d14578

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

qa/L0_jax_unittest/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ export XLA_FLAGS="${XLA_FLAGS} --xla_gpu_deterministic_ops"
3636
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"
3737
# Test without custom calls
3838
export 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

4141
if [ $RET -ne 0 ]; then
4242
echo "Error: some sub-tests failed: $FAILED_CASES"

0 commit comments

Comments
 (0)