Skip to content

Commit f883058

Browse files
huatuolihuatuoli
committed
[CI/CD] Update iluvatar3.6 ci runner.
1 parent 544ac0d commit f883058

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
@@ -72,7 +72,7 @@ jobs:
7272
7373
pip3 install "pytest==9.1.1"
7474
75-
for pkg in hypothesis absl-py scipy pytest-forked; do
75+
for pkg in hypothesis absl-py scipy pytest-forked expecttest; do
7676
pip3 list "$pkg" | grep "$pkg" || pip3 install "$pkg"
7777
done
7878
@@ -83,16 +83,77 @@ jobs:
8383
fi
8484
fi
8585
86-
# triton unit test
86+
# triton language unit test
8787
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/language/test_core.py
88+
pytest -v third_party/iluvatar/python/test/unit/language/test_annotations.py
89+
pytest -v third_party/iluvatar/python/test/unit/language/test_block_pointer.py
90+
pytest -v third_party/iluvatar/python/test/unit/language/test_compile_errors.py
91+
pytest -v third_party/iluvatar/python/test/unit/language/test_conversions.py
92+
pytest -v third_party/iluvatar/python/test/unit/language/test_decorator.py
93+
pytest -v third_party/iluvatar/python/test/unit/language/test_frontend.py
94+
pytest -v third_party/iluvatar/python/test/unit/language/test_libdevice.py
95+
pytest -v third_party/iluvatar/python/test/unit/language/test_iluvatar_math_fp16_bf16.py
96+
pytest -v third_party/iluvatar/python/test/unit/language/test_iluvatar_int8_upcast_dot_pipeline.py
97+
pytest -v third_party/iluvatar/python/test/unit/language/test_matmul.py
98+
pytest -v third_party/iluvatar/python/test/unit/language/test_mxfp.py
99+
pytest -v third_party/iluvatar/python/test/unit/language/test_pipeliner.py
100+
pytest -v third_party/iluvatar/python/test/unit/language/test_random.py
101+
pytest -v third_party/iluvatar/python/test/unit/language/test_reproducer.py
102+
pytest -v third_party/iluvatar/python/test/unit/language/test_standard.py
88103
pytest -v third_party/iluvatar/python/test/unit/language/test_subprocess.py
104+
UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/language/test_tensor_descriptor.py
105+
pytest -v third_party/iluvatar/python/test/unit/language/test_tuple.py
106+
pytest -v third_party/iluvatar/python/test/unit/language/test_warp_specialization.py
107+
108+
# triton operators unit test
89109
pytest -v third_party/iluvatar/python/test/unit/operators/test_blocksparse.py
90110
pytest -v third_party/iluvatar/python/test/unit/operators/test_cross_entropy.py
91111
pytest -v third_party/iluvatar/python/test/unit/operators/test_dot_trans.py
92112
pytest -v third_party/iluvatar/python/test/unit/operators/test_flash_attention.py
93113
pytest -v third_party/iluvatar/python/test/unit/operators/test_inductor.py
94114
pytest -v third_party/iluvatar/python/test/unit/operators/test_matmul.py
115+
pytest -v third_party/iluvatar/python/test/unit/operators/test_sme.py
116+
117+
# triton runtime unit test
118+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_autotuner.py
119+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_bindings.py
120+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_build.py
121+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_cache.py
122+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_compilation_listener.py
123+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_driver.py
124+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_launch.py
125+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_specialize.py
126+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_subproc.py
127+
pytest -v third_party/iluvatar/python/test/unit/runtime/test_iluvatar_loop_unroll_warning.py
128+
129+
# triton instrumentation unit test
130+
pytest -v third_party/iluvatar/python/test/unit/instrumentation/test_gpuhello.py
131+
132+
# triton tools unit test
133+
pytest -v third_party/iluvatar/python/test/unit/tools/test_irsource.py
134+
pytest -v third_party/iluvatar/python/test/unit/tools/test_linear_layout.py
135+
136+
# triton misc unit test
137+
pytest -v third_party/iluvatar/python/test/unit/test_debug_dump.py
95138
pytest -v third_party/iluvatar/python/test/unit/test_debug.py
139+
pytest -v third_party/iluvatar/python/test/unit/test_debuginfo.py
140+
pytest -v third_party/iluvatar/python/test/unit/test_filecheck.py
141+
pytest -v third_party/iluvatar/python/test/unit/test_knobs.py
142+
# UMD_CUDAMODULELOADING=0 pytest -v third_party/iluvatar/python/test/unit/test_link.py
143+
pytest -v third_party/iluvatar/python/test/unit/test_perf_warning.py
144+
145+
# triton regression test
146+
pytest -v third_party/iluvatar/python/test/regression/test_cast_matmul.py
147+
pytest -v third_party/iluvatar/python/test/regression/test_functional_regressions.py
148+
149+
# triton integrations unit test
150+
pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/chunk_o/test_chunk_fwd_kernel_o.py
151+
pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/wy_fast/test_recompute_w_u.py
152+
pytest -v third_party/iluvatar/python/test/unit/integrations/fbgemm/test_jagged_flash_attention_bwd_basic_min.py
153+
PUNICA_TEST_LEVEL=quick pytest -v third_party/iluvatar/python/test/unit/integrations/vllm/punica_lora/test_punica_ops.py
154+
pytest -v third_party/iluvatar/python/test/unit/integrations/sglang/flash_mla/test_flash_mla_ut.py
155+
pytest -v third_party/iluvatar/python/test/unit/integrations/inductor/test_bucketize_matmul.py
156+
pytest -v third_party/iluvatar/python/test/unit/integrations/inductor/test_swfw3103_flex_attention_precision.py
96157
97158
# TLE unit test
98159
pytest -v python/test/tle/integration/test_tle_local_store.py
@@ -101,6 +162,16 @@ jobs:
101162
pytest -v python/test/tle/unit/test_extract_tile_dynamic_index.py
102163
pytest -v python/test/tle/unit/test_insert_tile_static_index.py
103164
pytest -v python/test/tle/unit/test_insert_tile_dynamic_index.py
165+
pytest -v python/test/tle/unit/test_tle.py
166+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_copy.py
167+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_async_load.py
168+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_memory_space.py
169+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_cumsum.py
170+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipeline.py
171+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipeline_e2e.py
172+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_warp_specialize.py
173+
pytest -v third_party/iluvatar/python/test/unit/tle/test_tle_pipe.py
104174
105175
# TLE tutorials unit tests
106-
python3 python/tutorials/tle/01-fft.py --only_unit_test
176+
python3 third_party/iluvatar/python/tutorials/tle/01-fft.py --only_unit_test
177+
python3 third_party/iluvatar/python/tutorials/tle/03-topk.py --only_unit_test

0 commit comments

Comments
 (0)