Skip to content

Commit a39f49e

Browse files
author
zhoujiamei
committed
fixed format
1 parent aa45e58 commit a39f49e

3 files changed

Lines changed: 12 additions & 12 deletions

File tree

build_tools/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,4 +381,4 @@ def copy_common_headers(
381381
for path in headers:
382382
new_path = dst_dir / path.relative_to(src_dir)
383383
new_path.parent.mkdir(exist_ok=True, parents=True)
384-
shutil.copy(path, new_path)
384+
shutil.copy(path, new_path)

qa/L0_pytorch_debug_unittest/test.sh

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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" \
5454
run_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
5858
run_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
6262
run_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
6666
run_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
7777
run_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 \
7979
pytest -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
8282
run_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 \
8484
pytest -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

qa/L0_pytorch_unittest/test.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ run_test_step() {
2121
local cmd=$3
2222
local label=$4
2323

24-
24+
2525
if [ "$PLATFORM" = "metax" ]; then
2626
case "$test_path" in
2727
*"test_numerics.py" | \
@@ -45,10 +45,10 @@ run_test_step() {
4545
esac
4646
fi
4747

48-
48+
4949
echo "-------------------------------------------------------"
5050
echo "[RUN] Executing: $label"
51-
51+
5252
eval "$cmd" || test_fail "$label"
5353
}
5454

@@ -119,4 +119,4 @@ if [ "$FAIL" -ne 0 ]; then
119119
exit 1
120120
fi
121121
echo "All assigned tests passed (some might have been skipped)."
122-
exit 0
122+
exit 0

0 commit comments

Comments
 (0)