@@ -65,44 +65,44 @@ jobs:
6565 export ASCEND_VISIBLE_DEVICES=7
6666 export TORCH_DEVICE_BACKEND_AUTOLOAD=0
6767 cd python
68- MAX_JOBS=32 python3 -m pip install . --no-build-isolation
68+ MAX_JOBS=32 python3.11 -m pip install . --no-build-isolation
6969
7070 - name : FlagTree Test on Ascend
7171 if : steps.check_files.outputs.only_docs_changed != 'true'
7272 shell : bash
7373 run : |
7474 set -x
7575 source /usr/local/Ascend/ascend-toolkit/set_env.sh
76- python3 third_party/tests/ascend/vector-add.py
77- python3 third_party/ascend/examples/tutorials/01-vector-add.py
78- python3 third_party/ascend/examples/tutorials/02-fused-softmax.py
79- python3 third_party/ascend/examples/tutorials/03-layer-norm.py
80- python3 third_party/ascend/examples/tutorials/04-fused-attention.py
81- python3 third_party/ascend/examples/tutorials/06-demo-autotune.py
82- python3 third_party/ascend/examples/tutorials/07-profiler.py
83- python3 third_party/ascend/examples/tutorials/09-gather.py
84- python3 third_party/ascend/examples/tutorials/10-gather_sorted.py
85- python3 third_party/ascend/examples/tutorials/11-rab_time.py
86- python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized.py
87- python3 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized-flagtree.py
88- python3 third_party/ascend/examples/tutorials/14-accuracy-comparison.py
89- python3 python/test/ops/01_vector_add/01_vector_add.py
90- python3 python/test/ops/abs/abs.py
91- python3 python/test/ops/addmm/addmm.py
92- python3 python/test/ops/addmm/addmm_ascend.py
93- python3 python/test/ops/amax/amax.py
94- python3 python/test/ops/amax/amax_ascend_perf.py
95- python3 python/test/ops/apply_rotary_pos_emb/apply_rotary_pos_emb.py
96- python3 python/test/ops/apply_rotary_pos_emb/apply_rotary_pos_emb_ascend.py
97- python3 python/test/ops/argmin/argmin.py
98- python3 python/test/ops/argmin/argmin_ascend_perf.py
99- python3 python/test/ops/bmm/bmm_ascend.py
100- python3 python/test/ops/cumsum/cumsum.py
101- python3 python/test/ops/min_dim/min_dim.py
102- python3 python/test/ops/min_dim/min_dim_ascend_perf.py
103- python3 python/test/ops/sum_dim/sum_dim.py
104- python3 python/test/ops/varmean/var_mean_ascend.py
105- python3 -m pytest third_party/ascend/examples/pytest_ut --ignore=third_party/ascend/examples/pytest_ut/test_index_select.py \
76+ python3.11 third_party/tests/ascend/vector-add.py
77+ python3.11 third_party/ascend/examples/tutorials/01-vector-add.py
78+ python3.11 third_party/ascend/examples/tutorials/02-fused-softmax.py
79+ python3.11 third_party/ascend/examples/tutorials/03-layer-norm.py
80+ python3.11 third_party/ascend/examples/tutorials/04-fused-attention.py
81+ python3.11 third_party/ascend/examples/tutorials/06-demo-autotune.py
82+ python3.11 third_party/ascend/examples/tutorials/07-profiler.py
83+ python3.11 third_party/ascend/examples/tutorials/09-gather.py
84+ python3.11 third_party/ascend/examples/tutorials/10-gather_sorted.py
85+ python3.11 third_party/ascend/examples/tutorials/11-rab_time.py
86+ python3.11 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized.py
87+ python3.11 third_party/ascend/examples/tutorials/13-matrix-multiplication-optimized-flagtree.py
88+ python3.11 third_party/ascend/examples/tutorials/14-accuracy-comparison.py
89+ python3.11 python/test/ops/01_vector_add/01_vector_add.py
90+ python3.11 python/test/ops/abs/abs.py
91+ python3.11 python/test/ops/addmm/addmm.py
92+ python3.11 python/test/ops/addmm/addmm_ascend.py
93+ python3.11 python/test/ops/amax/amax.py
94+ python3.11 python/test/ops/amax/amax_ascend_perf.py
95+ python3.11 python/test/ops/apply_rotary_pos_emb/apply_rotary_pos_emb.py
96+ python3.11 python/test/ops/apply_rotary_pos_emb/apply_rotary_pos_emb_ascend.py
97+ python3.11 python/test/ops/argmin/argmin.py
98+ python3.11 python/test/ops/argmin/argmin_ascend_perf.py
99+ python3.11 python/test/ops/bmm/bmm_ascend.py
100+ python3.11 python/test/ops/cumsum/cumsum.py
101+ python3.11 python/test/ops/min_dim/min_dim.py
102+ python3.11 python/test/ops/min_dim/min_dim_ascend_perf.py
103+ python3.11 python/test/ops/sum_dim/sum_dim.py
104+ python3.11 python/test/ops/varmean/var_mean_ascend.py
105+ python3.11 -m pytest third_party/ascend/examples/pytest_ut --ignore=third_party/ascend/examples/pytest_ut/test_index_select.py \
106106 --ignore=third_party/ascend/examples/pytest_ut/test_linearize_permute.py \
107107 --ignore=third_party/ascend/examples/pytest_ut/test_logical_and.py \
108108 --ignore=third_party/ascend/examples/pytest_ut/test_logical_or.py \
0 commit comments