Aviral Goel 3e77721755 feat: add split_k support for block scale gemm bquant mode. (#3653) ...

* WIP: add splitk to bquant

* feat: add support for bf8i4 and fp8i4 by calculating correct stride for packed data types

* chore: remove temporary test script

* fix: incorrect tile window length for splitted bq tensor window

* chore: improve comments

* test: add unit tests to cover bquant splitk functionality

* fix: conflict resolution by renaming variables

2026-02-02 14:41:53 -08:00

..

gemm_quant_kernel.hpp

feat: add split_k support for block scale gemm bquant mode. (#3653)

2026-02-02 14:41:53 -08:00

grouped_gemm_quant_kernel.hpp

feat: add split_k support for block scale gemm bquant mode. (#3653)

2026-02-02 14:41:53 -08:00