KurisandClaude Opus 4.6 e45ecf7dc4 [Feature] Support T.annotate_compile_flags, T.annotate_pass_configs, and out_idx as PrimFunc attrs (#2006) ...

Allow configuring pass configs, compile flags, and out_idx directly
inside function bodies using T.annotate_compile_flags(),
T.annotate_pass_configs(), and T.empty()+return. These are stored as
proper PrimFunc attrs (tilelang_compile_flags, tilelang_pass_configs,
tilelang_out_idx) instead of monkey-patching, and merged at compile
time. Annotations can be placed before or after tensor type annotations.

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

2026-04-02 12:42:47 +08:00

..

test_tilelang_intimm.py

[Lint] Phaseout Yapf format and embrace ruff format (#1417)

2025-12-12 17:22:18 +08:00

test_tilelang_language_access_ptr_codegen.py

[Refactor] Improve cp.async lowering and add async_copy op (#1887)

2026-03-06 17:35:16 +08:00

test_tilelang_language_access_ptr.py

[Refactor] Introduce T.access_of to combine T.address_of and access_ptr (#1827)

2026-02-11 22:51:59 +08:00

test_tilelang_language_alias.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_all_of.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_alloc_var_cmp.py

[Bugfix] Fix alloc_var re-bind warning when assigned with comparison ops (#1974)

2026-03-27 11:17:21 +08:00

test_tilelang_language_alloc.py

[Feature] Support alloc global workspace (#1940)

2026-03-23 19:26:24 +08:00

test_tilelang_language_annot.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_annotate_safe_value.py

[Refactor] Improve cp.async lowering and add async_copy op (#1887)

2026-03-06 17:35:16 +08:00

test_tilelang_language_any_of.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_assume.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_async_copy_gemm_sm80.py

[Refactor] Improve cp.async lowering and add async_copy op (#1887)

2026-03-06 17:35:16 +08:00

test_tilelang_language_atomic.py

[BugFix] Fix type mismatch when lowering to AtomicAddx2 template (#1898)

2026-03-04 20:44:12 +08:00

test_tilelang_language_augassign.py

[Bugfix] Fix ast builder error for value -= 1 (#1825)

2026-02-09 17:30:22 +08:00

test_tilelang_language_ceildiv.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_chain_equal.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_clamp.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_clear.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_cluster.py

[Feature] Support cluster launch, query, synchronization and barrier operations (#1874)

2026-02-28 14:17:47 +08:00

test_tilelang_language_composable_index.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_cooperative.py

[Fix] Add register to read A ptr in test_tilelang_language_cooperative.py (#1593)

2026-01-04 15:45:38 +08:00

test_tilelang_language_copy.py

[Enhancement] Clarify the semantic rule of copy operator and add shape mismatched tests (#1883)

2026-02-27 19:49:38 +08:00

test_tilelang_language_cumsum.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_eager_jit.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_frontend_v2.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_func_attrs.py

[Feature] Support T.annotate_compile_flags, T.annotate_pass_configs, and out_idx as PrimFunc attrs (#2006)

2026-04-02 12:42:47 +08:00

test_tilelang_language_get_warp_info.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_if_range.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_infinity.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_int64.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_intrinsics_codegen.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_isfinite_codegen.py

[Feature] Add CUDA intrinsic for isfinite operation (#1925)

2026-03-12 17:21:01 +08:00

test_tilelang_language_ldg.py

[Refactor] Unify the usage of cast-related operators (#1757)

2026-02-02 18:47:18 +08:00

test_tilelang_language_let_layout.py

[Enhancement] Enhance let binding handling in layout inference and warp specialized pass (#1484)

2025-12-21 03:26:12 +08:00

test_tilelang_language_let.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_mask_op.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_min_blocks_per_sm.py

[Feature] Introduce annotation for minBlocksPerMultiprocessor in __launch_bounds__ (#1979)

2026-03-26 19:42:20 +08:00

test_tilelang_language_negative_index.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_parallel.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_pdl.py

[Bugfix] Fix pdl related intrin handling to avoid strict annotation codegen (#1650)

2026-01-09 17:31:33 +08:00

test_tilelang_language_pipeline.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_ptr.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_rand.py

[BugFix] Stride check and fix for tensors with zero-stride argument (#1749)

2026-01-29 23:59:24 +08:00

test_tilelang_language_reduce.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_reshape.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_select.py

[Refactor] Unify the usage of cast-related operators (#1757)

2026-02-02 18:47:18 +08:00

test_tilelang_language_stg.py

[Refactor] Unify the usage of cast-related operators (#1757)

2026-02-02 18:47:18 +08:00

test_tilelang_language_subtype.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_tcgen05_gemm.py

[Refactor] Separate gemm into explicit wgmma_gemm and tcgen05_gemm functions (#1949)

2026-03-20 14:54:43 +08:00

test_tilelang_language_ternary.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_tma_1d.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_tma_copy.py

[Feature] Support TMA store in T.tma_copy() (#1981)

2026-03-27 15:29:48 +08:00

test_tilelang_language_tma_store.py

[Feature] Support TMA store in T.tma_copy() (#1981)

2026-03-27 15:29:48 +08:00

test_tilelang_language_transpose.py

Introduce T.deallocate_tmem and T.transpose (#1971)

2026-03-30 15:56:21 +08:00

test_tilelang_language_unroll.py

[Enhancement] Improve unroll loop functionality for dynamic extent and corresponding test case (#1654)

2026-01-12 22:33:17 +08:00

test_tilelang_language_var_init.py

[Bugfix] Open Rocm ci test and fix some bugs. (#1443)

2026-01-06 02:18:29 +08:00

test_tilelang_language_vectorize.py

test: reduce CI runtime for slow Python suites (#1932)

2026-03-18 13:08:30 +08:00

test_tilelang_language_vectorized_cast.py

[BugFix] Fix FP4 related vectorized cast (#1741)

2026-01-27 20:54:40 +08:00

test_tilelang_language_view.py

Support packed subtype views during layout reshape (#1947)

2026-03-19 16:08:15 +08:00

test_tilelang_language_warp_reduce.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_language_warp_sync.py

[Feature] add T.sync_warp & T.shfl_sync; change extern pdl into intrin (#1614)

2026-01-06 23:54:11 +08:00

test_tilelang_language_wgmma_gemm.py

[Refactor] Separate gemm into explicit wgmma_gemm and tcgen05_gemm functions (#1949)

2026-03-20 14:54:43 +08:00

test_tilelang_memory_leak.py

[Testing] Add Memory Leak Test (#1516)

2025-12-24 14:19:39 +08:00