9292 TE_PATH : .
9393 run : |
9494 bash ./qa/L0_pytorch_lint/test.sh
95- echo "\n-----------------------------------------------------"
95+
96+ echo ""
97+ echo "-----------------------------------------------------"
9698 echo "Note: Pylint check ignores errors C0411 (incorrect import position) and W0611 (unused import), which can be achieved by adding the parameter --disable=C0411,W0611"
9799 echo "-----------------------------------------------------"
98100 continue-on-error : true
@@ -104,7 +106,9 @@ jobs:
104106 TE_PATH : .
105107 run : |
106108 bash ./qa/L0_pytorch_lint/test.sh
107- echo "\n-----------------------------------------------------"
109+
110+ echo ""
111+ echo "-----------------------------------------------------"
108112 echo "Note: Pylint check ignores errors C0411 (incorrect import position) and W0611 (unused import), which can be achieved by adding the parameter --disable=C0411,W0611"
109113 echo "-----------------------------------------------------"
110114 continue-on-error : true
@@ -113,38 +117,19 @@ jobs:
113117 run : |
114118 source .github/workflows/scripts/gpu_check.sh
115119 wait_for_gpu
116-
117- # - name: L0_cppunittest
118- # working-directory: tests/cpp
119- # run: |
120- # TE_LIB_PATH=$(pip3 show transformer-engine | grep -E "Location:|Editable project location:" | tail -n 1 | awk '{print $NF}')
121- # export LD_LIBRARY_PATH=$TE_LIB_PATH:$LD_LIBRARY_PATH
122-
123- # # Set parallelization parameters
124- # NUM_PHYSICAL_CORES=$(nproc)
125- # NUM_PARALLEL_JOBS=4
126-
127- # cmake -GNinja \
128- # -Bbuild \
129- # -DTE_LIB="${TE_LIB_FULL_PATH}" \
130- # .
131- # cmake --build build --verbose
132- # export OMP_NUM_THREADS=$((NUM_PHYSICAL_CORES / NUM_PARALLEL_JOBS))
133- # ctest --test-dir build -j$NUM_PARALLEL_JOBS
134-
135-
136-
120+
137121 - name : L0 Pytorch Debug Unittest
138122 id : L0_pytorch_debug_unittest
139123 continue-on-error : true
140124 env :
141125 XML_LOG_DIR : " /logs/pytorch/debug"
142126 TE_PATH : .
143127 run : |
144- ignore_files=("test_api_features.py" "test_numerics.py")
128+ ignore_files=("tests/pytorch/debug/ test_api_features.py" "tests/pytorch/ test_numerics.py")
145129
146130 for file in "${ignore_files[@]}"; do
147- sed -i 's/^.*${file}.*$/#&/' ./qa/L0_pytorch_debug_unittest/test.sh
131+ escaped_file=$(echo "$file" | sed 's/\//\\\//g')
132+ sed -i "s/^.*\($escaped_file\).*$/#&/" ./qa/L0_pytorch_debug_unittest/test.sh
148133 done
149134
150135 cat ./qa/L0_pytorch_debug_unittest/test.sh
@@ -158,10 +143,13 @@ jobs:
158143 XML_LOG_DIR : " /logs/pytorch"
159144 TE_PATH : .
160145 run : |
161- ignore_files=("test_numerics.py" "test_permutation.py" "test_attention.py" "test_checkpoint.py")
146+ ignore_files=("tests/pytorch/test_numerics.py" "tests/pytorch/test_permutation.py"
147+ "tests/pytorch/attention/test_attention.py" "tests/pytorch/test_checkpoint.py"
148+ )
162149
163150 for file in "${ignore_files[@]}"; do
164- sed -i 's/^.*${file}.*$/#&/' ./qa/L0_pytorch_unittest/test.sh
151+ escaped_file=$(echo "$file" | sed 's/\//\\\//g')
152+ sed -i 's/^.*\($file\).*$/#&/' ./qa/L0_pytorch_unittest/test.sh
165153 done
166154
167155 cat ./qa/L0_pytorch_unittest/test.sh
0 commit comments