Skip to content

Commit 5ed4121

Browse files
huatuolihuatuoli
committed
[CI/CD] Update iluvatar3.6 ci runner.
1 parent e8b67c9 commit 5ed4121

1 file changed

Lines changed: 74 additions & 3 deletions

File tree

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

Lines changed: 74 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ jobs:
8484
8585
pip3 install "pytest==9.1.1"
8686
87-
for pkg in hypothesis absl-py scipy pytest-forked; do
87+
for pkg in hypothesis absl-py scipy pytest-forked expecttest; do
8888
pip3 list "$pkg" | grep "$pkg" || pip3 install "$pkg"
8989
done
9090
@@ -95,16 +95,77 @@ jobs:
9595
fi
9696
fi
9797
98-
# triton unit test
98+
# triton language unit test
9999
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/language/test_core.py
100+
pytest -v third_party/iluvatar/python/test/unit/language/test_annotations.py
101+
pytest -v third_party/iluvatar/python/test/unit/language/test_block_pointer.py
102+
pytest -v third_party/iluvatar/python/test/unit/language/test_compile_errors.py
103+
pytest -v third_party/iluvatar/python/test/unit/language/test_conversions.py
104+
pytest -v third_party/iluvatar/python/test/unit/language/test_decorator.py
105+
pytest -v third_party/iluvatar/python/test/unit/language/test_frontend.py
106+
pytest -v third_party/iluvatar/python/test/unit/language/test_libdevice.py
107+
pytest -v third_party/iluvatar/python/test/unit/language/test_iluvatar_math_fp16_bf16.py
108+
pytest -v third_party/iluvatar/python/test/unit/language/test_iluvatar_int8_upcast_dot_pipeline.py
109+
pytest -v third_party/iluvatar/python/test/unit/language/test_matmul.py
110+
pytest -v third_party/iluvatar/python/test/unit/language/test_mxfp.py
111+
pytest -v third_party/iluvatar/python/test/unit/language/test_pipeliner.py
112+
pytest -v third_party/iluvatar/python/test/unit/language/test_random.py
113+
pytest -v third_party/iluvatar/python/test/unit/language/test_reproducer.py
114+
pytest -v third_party/iluvatar/python/test/unit/language/test_standard.py
100115
pytest -v third_party/iluvatar/python/test/unit/language/test_subprocess.py
116+
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/language/test_tensor_descriptor.py
117+
pytest -v third_party/iluvatar/python/test/unit/language/test_tuple.py
118+
pytest -v third_party/iluvatar/python/test/unit/language/test_warp_specialization.py
119+
120+
# triton operators unit test
101121
pytest -v third_party/iluvatar/python/test/unit/operators/test_blocksparse.py
102122
pytest -v third_party/iluvatar/python/test/unit/operators/test_cross_entropy.py
103123
pytest -v third_party/iluvatar/python/test/unit/operators/test_dot_trans.py
104124
pytest -v third_party/iluvatar/python/test/unit/operators/test_flash_attention.py
105125
pytest -v third_party/iluvatar/python/test/unit/operators/test_inductor.py
106126
pytest -v third_party/iluvatar/python/test/unit/operators/test_matmul.py
127+
pytest -v third_party/iluvatar/python/test/unit/operators/test_sme.py
128+
129+
# triton runtime unit test
130+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_autotuner.py
131+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_bindings.py
132+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_build.py
133+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_cache.py
134+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_compilation_listener.py
135+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_driver.py
136+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_launch.py
137+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_specialize.py
138+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_subproc.py
139+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_iluvatar_loop_unroll_warning.py
140+
141+
# triton instrumentation unit test
142+
pytest -v third_party/iluvatar/python/test/unit/instrumentation/test_gpuhello.py
143+
144+
# triton tools unit test
145+
pytest -v third_party/iluvatar/python/test/unit/tools/test_irsource.py
146+
pytest -v third_party/iluvatar/python/test/unit/tools/test_linear_layout.py
147+
148+
# triton misc unit test
149+
pytest -v third_party/iluvatar/python/test/unit/test_debug_dump.py
107150
pytest -v third_party/iluvatar/python/test/unit/test_debug.py
151+
pytest -v third_party/iluvatar/python/test/unit/test_debuginfo.py
152+
pytest -v third_party/iluvatar/python/test/unit/test_filecheck.py
153+
pytest -v third_party/iluvatar/python/test/unit/test_knobs.py
154+
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/test_link.py
155+
pytest -v third_party/iluvatar/python/test/unit/test_perf_warning.py
156+
157+
# triton regression test
158+
pytest -v third_party/iluvatar/python/test/regression/test_cast_matmul.py
159+
pytest -v third_party/iluvatar/python/test/regression/test_functional_regressions.py
160+
161+
# triton integrations unit test
162+
pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/chunk_o/test_chunk_fwd_kernel_o.py
163+
pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/wy_fast/test_recompute_w_u.py
164+
pytest -v third_party/iluvatar/python/test/unit/integrations/fbgemm/test_jagged_flash_attention_bwd_basic_min.py
165+
PUNICA_TEST_LEVEL=quick pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/punica_lora/test_punica_ops.py
166+
pytest -v third_party/iluvatar/python/test/unit/integrations/sglang/flash_mla/test_flash_mla_ut.py
167+
pytest -v third_party/iluvatar/python/test/unit/integrations/inductor/test_bucketize_matmul.py
168+
pytest -v third_party/iluvatar/python/test/unit/integrations/inductor/test_swfw3103_flex_attention_precision.py
108169
109170
# TLE unit test
110171
pytest -v python/test/tle/integration/test_tle_local_store.py
@@ -113,6 +174,16 @@ jobs:
113174
pytest -v python/test/tle/unit/test_extract_tile_dynamic_index.py
114175
pytest -v python/test/tle/unit/test_insert_tile_static_index.py
115176
pytest -v python/test/tle/unit/test_insert_tile_dynamic_index.py
177+
pytest -v python/test/tle/unit/test_tle.py
178+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_copy.py
179+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_async_load.py
180+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_memory_space.py
181+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_cumsum.py
182+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipeline.py
183+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipeline_e2e.py
184+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_warp_specialize.py
185+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipe.py
116186
117187
# TLE tutorials unit tests
118-
python3 python/tutorials/tle/01-fft.py --only_unit_test
188+
python3 third_party/iluvatar/python/tutorials/tle/01-fft.py --only_unit_test
189+
python3 third_party/iluvatar/python/tutorials/tle/03-topk.py --only_unit_test

0 commit comments

Comments
 (0)