tilelang

mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-21 06:40:23 +08:00

Files

T

Lei Wang 9c95a42dfb [Refactor] Phaseout legacy util map_torch_type with T.dtype.as_torch (#2075 )

* Remove redundant `map_torch_type` in favor of `T.dtype.as_torch()`

`map_torch_type` was a standalone function that converted dtype strings or
TVM DataType objects to `torch.dtype`. This was redundant since
`T.dtype.as_torch()` (and `T.dtype(x).as_torch()`) already provides the
same functionality with better handling (HIP/CUDA awareness, proper dtype
lookup tables, etc.).

Changes:
- Remove `map_torch_type` from `tilelang/utils/tensor.py` and its re-export
- Make `T.dtype()` idempotent for existing dtype objects (pass-through)
- Add backward-compat for `"e4m3fnuz_float8"` TVM internal name in `as_torch()`
- Replace all usages across production code, tests, and examples
- Rebuild Cython extension to reflect the import change

* Refactor output and accumulation data types in GEMM FP8 example

Updated the `out_dtype` and `accum_dtype` variables in the `example_tilelang_gemm_fp8_intrinsic.py` file to use `T.float32` instead of string literals. This change enhances type consistency and aligns with recent updates in type handling.

2026-04-21 18:27:15 +08:00

test_tilelang_kernel_bf16_gemm_mma.py

[Refactor] Phaseout legacy util map_torch_type with T.dtype.as_torch (#2075 )

2026-04-21 18:27:15 +08:00

test_tilelang_kernel_bf16_gemm_tcgen5_ts.py

[Refactor][Pipeline] Run pipeline rewriting before layout inference and stabilize tiled WS (#2002 )

2026-04-07 23:54:55 +08:00

test_tilelang_kernel_element_wise_add.py

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

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

test_tilelang_kernel_fp8_gemm_mma.py

[Refactor] Phaseout legacy util map_torch_type with T.dtype.as_torch (#2075 )