Skip to content

Commit 1c43832

Browse files
authored
[DOC][CI] Add spacemit to readme, and update workflow (#1004)
1 parent a484d49 commit 1c43832

5 files changed

Lines changed: 93 additions & 77 deletions

File tree

.github/workflows/iluvatar3.6-build-and-test.yml

Lines changed: 71 additions & 73 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,6 @@ jobs:
8787
set -x
8888
source ~/env.sh
8989
90-
pip3 install "pytest==9.1.1"
91-
9290
for pkg in hypothesis absl-py scipy pytest-forked expecttest; do
9391
pip3 list "$pkg" | grep "$pkg" || pip3 install "$pkg"
9492
done
@@ -101,93 +99,93 @@ jobs:
10199
fi
102100
103101
# triton language unit test
104-
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/language/test_core.py
105-
pytest -v third_party/iluvatar/python/test/unit/language/test_annotations.py
106-
pytest -v third_party/iluvatar/python/test/unit/language/test_block_pointer.py
107-
pytest -v third_party/iluvatar/python/test/unit/language/test_compile_errors.py
108-
pytest -v third_party/iluvatar/python/test/unit/language/test_conversions.py
109-
pytest -v third_party/iluvatar/python/test/unit/language/test_decorator.py
110-
pytest -v third_party/iluvatar/python/test/unit/language/test_frontend.py
111-
pytest -v third_party/iluvatar/python/test/unit/language/test_libdevice.py
112-
pytest -v third_party/iluvatar/python/test/unit/language/test_iluvatar_math_fp16_bf16.py
113-
pytest -v third_party/iluvatar/python/test/unit/language/test_iluvatar_int8_upcast_dot_pipeline.py
114-
pytest -v third_party/iluvatar/python/test/unit/language/test_matmul.py
115-
pytest -v third_party/iluvatar/python/test/unit/language/test_mxfp.py
116-
pytest -v third_party/iluvatar/python/test/unit/language/test_pipeliner.py
117-
pytest -v third_party/iluvatar/python/test/unit/language/test_random.py
118-
pytest -v third_party/iluvatar/python/test/unit/language/test_reproducer.py
119-
pytest -v third_party/iluvatar/python/test/unit/language/test_standard.py
120-
pytest -v third_party/iluvatar/python/test/unit/language/test_subprocess.py
121-
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/language/test_tensor_descriptor.py
122-
pytest -v third_party/iluvatar/python/test/unit/language/test_tuple.py
123-
pytest -v third_party/iluvatar/python/test/unit/language/test_warp_specialization.py
102+
UMD_CUDAMODULELOADING=0 pytest third_party/iluvatar/python/test/unit/language/test_core.py
103+
pytest third_party/iluvatar/python/test/unit/language/test_annotations.py
104+
pytest third_party/iluvatar/python/test/unit/language/test_block_pointer.py
105+
pytest third_party/iluvatar/python/test/unit/language/test_compile_errors.py
106+
pytest third_party/iluvatar/python/test/unit/language/test_conversions.py
107+
pytest third_party/iluvatar/python/test/unit/language/test_decorator.py
108+
pytest third_party/iluvatar/python/test/unit/language/test_frontend.py
109+
pytest third_party/iluvatar/python/test/unit/language/test_libdevice.py
110+
pytest third_party/iluvatar/python/test/unit/language/test_iluvatar_math_fp16_bf16.py
111+
pytest third_party/iluvatar/python/test/unit/language/test_iluvatar_int8_upcast_dot_pipeline.py
112+
pytest third_party/iluvatar/python/test/unit/language/test_matmul.py
113+
pytest third_party/iluvatar/python/test/unit/language/test_mxfp.py
114+
pytest third_party/iluvatar/python/test/unit/language/test_pipeliner.py
115+
pytest third_party/iluvatar/python/test/unit/language/test_random.py
116+
pytest third_party/iluvatar/python/test/unit/language/test_reproducer.py
117+
pytest third_party/iluvatar/python/test/unit/language/test_standard.py
118+
pytest third_party/iluvatar/python/test/unit/language/test_subprocess.py
119+
UMD_CUDAMODULELOADING=0 pytest third_party/iluvatar/python/test/unit/language/test_tensor_descriptor.py
120+
pytest third_party/iluvatar/python/test/unit/language/test_tuple.py
121+
pytest third_party/iluvatar/python/test/unit/language/test_warp_specialization.py
124122
125123
# triton operators unit test
126-
pytest -v third_party/iluvatar/python/test/unit/operators/test_blocksparse.py
127-
pytest -v third_party/iluvatar/python/test/unit/operators/test_cross_entropy.py
128-
pytest -v third_party/iluvatar/python/test/unit/operators/test_dot_trans.py
129-
pytest -v third_party/iluvatar/python/test/unit/operators/test_flash_attention.py
130-
pytest -v third_party/iluvatar/python/test/unit/operators/test_inductor.py
131-
pytest -v third_party/iluvatar/python/test/unit/operators/test_matmul.py
132-
pytest -v third_party/iluvatar/python/test/unit/operators/test_sme.py
124+
pytest third_party/iluvatar/python/test/unit/operators/test_blocksparse.py
125+
pytest third_party/iluvatar/python/test/unit/operators/test_cross_entropy.py
126+
pytest third_party/iluvatar/python/test/unit/operators/test_dot_trans.py
127+
pytest third_party/iluvatar/python/test/unit/operators/test_flash_attention.py
128+
pytest third_party/iluvatar/python/test/unit/operators/test_inductor.py
129+
pytest third_party/iluvatar/python/test/unit/operators/test_matmul.py
130+
pytest third_party/iluvatar/python/test/unit/operators/test_sme.py
133131
134132
# triton runtime unit test
135-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_autotuner.py
136-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_bindings.py
137-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_build.py
138-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_cache.py
139-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_compilation_listener.py
140-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_driver.py
141-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_launch.py
142-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_specialize.py
143-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_subproc.py
144-
pytest -v third_party/iluvatar/python/test/unit/runtime/test_iluvatar_loop_unroll_warning.py
133+
pytest third_party/iluvatar/python/test/unit/runtime/test_autotuner.py
134+
pytest third_party/iluvatar/python/test/unit/runtime/test_bindings.py
135+
pytest third_party/iluvatar/python/test/unit/runtime/test_build.py
136+
pytest third_party/iluvatar/python/test/unit/runtime/test_cache.py
137+
pytest third_party/iluvatar/python/test/unit/runtime/test_compilation_listener.py
138+
pytest third_party/iluvatar/python/test/unit/runtime/test_driver.py
139+
pytest third_party/iluvatar/python/test/unit/runtime/test_launch.py
140+
pytest third_party/iluvatar/python/test/unit/runtime/test_specialize.py
141+
pytest third_party/iluvatar/python/test/unit/runtime/test_subproc.py
142+
pytest third_party/iluvatar/python/test/unit/runtime/test_iluvatar_loop_unroll_warning.py
145143
146144
# triton instrumentation unit test
147-
pytest -v third_party/iluvatar/python/test/unit/instrumentation/test_gpuhello.py
145+
pytest third_party/iluvatar/python/test/unit/instrumentation/test_gpuhello.py
148146
149147
# triton tools unit test
150-
pytest -v third_party/iluvatar/python/test/unit/tools/test_irsource.py
151-
pytest -v third_party/iluvatar/python/test/unit/tools/test_linear_layout.py
148+
pytest third_party/iluvatar/python/test/unit/tools/test_irsource.py
149+
pytest third_party/iluvatar/python/test/unit/tools/test_linear_layout.py
152150
153151
# triton misc unit test
154-
pytest -v third_party/iluvatar/python/test/unit/test_debug_dump.py
155-
pytest -v third_party/iluvatar/python/test/unit/test_debug.py
156-
pytest -v third_party/iluvatar/python/test/unit/test_debuginfo.py
157-
pytest -v third_party/iluvatar/python/test/unit/test_filecheck.py
158-
pytest -v third_party/iluvatar/python/test/unit/test_knobs.py
159-
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/test_link.py
160-
pytest -v third_party/iluvatar/python/test/unit/test_perf_warning.py
152+
pytest third_party/iluvatar/python/test/unit/test_debug_dump.py
153+
pytest third_party/iluvatar/python/test/unit/test_debug.py
154+
pytest third_party/iluvatar/python/test/unit/test_debuginfo.py
155+
pytest third_party/iluvatar/python/test/unit/test_filecheck.py
156+
pytest third_party/iluvatar/python/test/unit/test_knobs.py
157+
UMD_CUDAMODULELOADING=0 pytest third_party/iluvatar/python/test/unit/test_link.py
158+
pytest third_party/iluvatar/python/test/unit/test_perf_warning.py
161159
162160
# triton regression test
163-
pytest -v third_party/iluvatar/python/test/regression/test_cast_matmul.py
164-
pytest -v third_party/iluvatar/python/test/regression/test_functional_regressions.py
161+
pytest third_party/iluvatar/python/test/regression/test_cast_matmul.py
162+
pytest third_party/iluvatar/python/test/regression/test_functional_regressions.py
165163
166164
# triton integrations unit test
167-
pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/chunk_o/test_chunk_fwd_kernel_o.py
168-
pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/wy_fast/test_recompute_w_u.py
169-
pytest -v third_party/iluvatar/python/test/unit/integrations/fbgemm/test_jagged_flash_attention_bwd_basic_min.py
170-
PUNICA_TEST_LEVEL=quick pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/punica_lora/test_punica_ops.py
171-
pytest -v third_party/iluvatar/python/test/unit/integrations/sglang/flash_mla/test_flash_mla_ut.py
172-
pytest -v third_party/iluvatar/python/test/unit/integrations/inductor/test_bucketize_matmul.py
173-
pytest -v third_party/iluvatar/python/test/unit/integrations/inductor/test_swfw3103_flex_attention_precision.py
165+
pytest third_party/iluvatar/python/test/unit/integrations/vllm/chunk_o/test_chunk_fwd_kernel_o.py
166+
pytest third_party/iluvatar/python/test/unit/integrations/vllm/wy_fast/test_recompute_w_u.py
167+
pytest third_party/iluvatar/python/test/unit/integrations/fbgemm/test_jagged_flash_attention_bwd_basic_min.py
168+
PUNICA_TEST_LEVEL=quick pytest third_party/iluvatar/python/test/unit/integrations/vllm/punica_lora/test_punica_ops.py
169+
pytest third_party/iluvatar/python/test/unit/integrations/sglang/flash_mla/test_flash_mla_ut.py
170+
pytest third_party/iluvatar/python/test/unit/integrations/inductor/test_bucketize_matmul.py
171+
pytest third_party/iluvatar/python/test/unit/integrations/inductor/test_swfw3103_flex_attention_precision.py
174172
175173
# TLE unit test
176-
pytest -v python/test/tle/integration/test_tle_local_store.py
177-
pytest -v python/test/tle/unit/test_tle_gpu_local_ptr.py
178-
pytest -v python/test/tle/unit/test_extract_tile_static_index.py
179-
pytest -v python/test/tle/unit/test_extract_tile_dynamic_index.py
180-
pytest -v python/test/tle/unit/test_insert_tile_static_index.py
181-
pytest -v python/test/tle/unit/test_insert_tile_dynamic_index.py
182-
pytest -v python/test/tle/unit/test_tle.py
183-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_copy.py
184-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_async_load.py
185-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_memory_space.py
186-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_cumsum.py
187-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipeline.py
188-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipeline_e2e.py
189-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_warp_specialize.py
190-
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipe.py
174+
pytest python/test/tle/integration/test_tle_local_store.py
175+
pytest python/test/tle/unit/test_tle_gpu_local_ptr.py
176+
pytest python/test/tle/unit/test_extract_tile_static_index.py
177+
pytest python/test/tle/unit/test_extract_tile_dynamic_index.py
178+
pytest python/test/tle/unit/test_insert_tile_static_index.py
179+
pytest python/test/tle/unit/test_insert_tile_dynamic_index.py
180+
pytest python/test/tle/unit/test_tle.py
181+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_copy.py
182+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_async_load.py
183+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_memory_space.py
184+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_cumsum.py
185+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_pipeline.py
186+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_pipeline_e2e.py
187+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_warp_specialize.py
188+
pytest third_party/iluvatar/python/test/unit/tle/test_tle_pipe.py
191189
192190
# TLE tutorials unit tests
193191
python3 third_party/iluvatar/python/tutorials/tle/01-fft.py --only_unit_test

.github/workflows/mthreads3.6-qwen-benchmark.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,14 @@ jobs:
8686
cd ~/flagrelease/qwen3.6
8787
bash ${BENCH_SCRIPT_DIR}/start.sh
8888
89+
- name: Test
90+
if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
91+
shell: bash
92+
run: |
93+
set -x
94+
cd ~/flagrelease/qwen3.6
95+
cat ${BENCH_SCRIPT_DIR}/test.sh
96+
8997
- name: Show the baseline
9098
if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
9199
shell: bash

.github/workflows/nvidia3.6-qwen-benchmark.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,14 @@ jobs:
8888
cd ~/flagrelease/qwen3.6
8989
bash ${BENCH_SCRIPT_DIR}/start.sh
9090
91+
- name: Test
92+
if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
93+
shell: bash
94+
run: |
95+
set -x
96+
cd ~/flagrelease/qwen3.6
97+
cat ${BENCH_SCRIPT_DIR}/test.sh
98+
9199
- name: Show the baseline
92100
if: ${{ steps.check_backend.outputs.should_skip != 'true' }}
93101
shell: bash

0 commit comments

Comments
 (0)