examples/dequantize_gemm/example_dequant_gemm_w4a8.py · 4bcb1593d4b4733696b3e1c764d0ff3c8320a5cf · OpenDAS / tilelang

[Example] add w4a8 gemm kernel (#815) · 4bcb1593

Cunxiao Ni authored Sep 16, 2025

* [Bugfix] fix autotune bug

* [Example] add w4a8 gemm kernel

* fix lint: pinned the version of `ml_dtypes`
The version of ml_dtypes should be pinned in the dependency specification. If the version of ml_dtypes is too low, it may result in errors such as fp4 not being defined.

* Renames example for dequantization GEMM

* format

* add w4a8 example to ci

* fix lint

4bcb1593

example_dequant_gemm_w4a8.py 7.4 KB

Replace example_dequant_gemm_w4a8.py