@@ -26,7 +26,7 @@ run_test_step() {
2626 local test_path=$2
2727 local cmd=$3
2828
29-
29+
3030 if [ " $PLATFORM " = " metax" ]; then
3131 case " $test_path " in
3232 * " test_numerics.py" | * " test_api_features.py" | * " test_sanity.py" )
@@ -38,7 +38,7 @@ run_test_step() {
3838 esac
3939 fi
4040
41-
41+
4242 echo " -------------------------------------------------------"
4343 echo " [RUN] Executing: $test_path "
4444 eval " $cmd " || FAIL=1
@@ -54,15 +54,15 @@ run_test_step "test_sanity.xml" "$TE_PATH/tests/pytorch/debug/test_sanity.py" \
5454run_test_step " test_config.xml" " $TE_PATH /tests/pytorch/debug/test_config.py" \
5555" pytest -v -s --junitxml=$XML_LOG_DIR /test_config.xml $TE_PATH /tests/pytorch/debug/test_config.py --feature_dirs=$NVTE_TEST_NVINSPECT_FEATURE_DIRS "
5656
57- # Step 3: Numerics
57+ # Step 3: Numerics
5858run_test_step " test_numerics.xml" " $TE_PATH /tests/pytorch/debug/test_numerics.py" \
5959" pytest -v -s --junitxml=$XML_LOG_DIR /test_numerics.xml $TE_PATH /tests/pytorch/debug/test_numerics.py --feature_dirs=$NVTE_TEST_NVINSPECT_FEATURE_DIRS "
6060
6161# Step 4: Log
6262run_test_step " test_log.xml" " $TE_PATH /tests/pytorch/debug/test_log.py" \
6363" pytest -v -s --junitxml=$XML_LOG_DIR /test_log.xml $TE_PATH /tests/pytorch/debug/test_log.py --feature_dirs=$NVTE_TEST_NVINSPECT_FEATURE_DIRS --configs_dir=$NVTE_TEST_NVINSPECT_CONFIGS_DIR "
6464
65- # Step 5: API Features
65+ # Step 5: API Features
6666run_test_step " test_api_features.xml" " $TE_PATH /tests/pytorch/debug/test_api_features.py" \
6767" NVTE_TORCH_COMPILE=0 pytest -v -s --junitxml=$XML_LOG_DIR /test_api_features.xml $TE_PATH /tests/pytorch/debug/test_api_features.py -k \" not (test_per_tensor_scaling or test_fake_quant or test_statistics_collection or test_statistics_multi_run)\" --no-header --feature_dirs=$NVTE_TEST_NVINSPECT_FEATURE_DIRS --configs_dir=$NVTE_TEST_NVINSPECT_CONFIGS_DIR "
6868
@@ -73,14 +73,14 @@ run_test_step "test_perf.xml" "$TE_PATH/tests/pytorch/debug/test_perf.py" \
7373
7474
7575
76- # Step 7: Sanity 2
76+ # Step 7: Sanity 2
7777run_test_step " test_sanity_2.xml" " $TE_PATH /tests/pytorch/test_sanity.py" \
7878" NVTE_TEST_NVINSPECT_ENABLED=1 NVTE_TEST_NVINSPECT_CONFIG_FILE=$NVTE_TEST_NVINSPECT_DUMMY_CONFIG_FILE NVTE_TEST_NVINSPECT_FEATURE_DIRS=$NVTE_TEST_NVINSPECT_FEATURE_DIRS PYTORCH_JIT=0 NVTE_TORCH_COMPILE=0 NVTE_ALLOW_NONDETERMINISTIC_ALGO=0 \
7979pytest -v -s --junitxml=$XML_LOG_DIR /test_sanity_2.xml $TE_PATH /tests/pytorch/test_sanity.py -k \" not (test_sanity_grouped_linear or test_inference_mode)\" --no-header"
8080
81- # Step 8: Numerics 2
81+ # Step 8: Numerics 2
8282run_test_step " test_numerics_2.xml" " $TE_PATH /tests/pytorch/test_numerics.py" \
8383" NVTE_TEST_NVINSPECT_ENABLED=1 NVTE_TEST_NVINSPECT_CONFIG_FILE=$NVTE_TEST_NVINSPECT_DUMMY_CONFIG_FILE NVTE_TEST_NVINSPECT_FEATURE_DIRS=$NVTE_TEST_NVINSPECT_FEATURE_DIRS PYTORCH_JIT=0 NVTE_TORCH_COMPILE=0 NVTE_ALLOW_NONDETERMINISTIC_ALGO=0 NVTE_FUSED_ATTN=0 \
8484pytest -v -s --junitxml=$XML_LOG_DIR /test_numerics_2.xml $TE_PATH /tests/pytorch/test_numerics.py -k \" not (test_linear_accuracy or test_layernorm_linear_accuracy or test_layernorm_mlp_accuracy or test_grouped_linear_accuracy or test_transformer_layer_hidden_states_format or test_grouped_gemm)\" --no-header"
8585
86- exit $FAIL
86+ exit $FAIL
0 commit comments