trending/tilelang
1
0
Fork 0
mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-22 06:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Address PassContext contamination from CI and fix incorrect rewrites in warp specialized pass (#767)

Browse Source 
* fix ci and pass bug

* fix

* try

* lint
This commit is contained in:
Wenhao Xie
2025-08-28 23:46:26 +08:00
committed by GitHub
parent 37051417b7
commit ff35fc0881
3 changed files with 26 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
testing/python/webgpu/test_webgpu_codegen.py
+3 -2
View File
@@ -43,8 +43,9 @@ def assert_gemm_codegen(
accum_dtype="float",
):
func = matmul(M, N, K, block_M, block_N, block_K, dtype=dtype, accum_dtype=accum_dtype)
artifact = tilelang.lower(func, target="webgpu")
# Because the current pass context have been polluted by previous testing.
with tvm.transform.PassContext():
artifact = tilelang.lower(func, target="webgpu")
src_code = artifact.kernel_source