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

..

CMakeLists.txt

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

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

test_gemm_quant_abquant_a4w4_base.cpp

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

test_gemm_quant_abquant_a4w4_padding.cpp

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

test_gemm_quant_abquant_a4w4_preshuffle.cpp

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

test_gemm_quant_abquant_base.cpp

Enable padding blockscale for abquant (#3453)

2025-12-24 09:12:40 -08:00

test_gemm_quant_abquant_padding.cpp

Enable padding blockscale for abquant (#3453)

2025-12-24 09:12:40 -08:00

test_gemm_quant_abquant_preshuffle_2d.cpp

[CK_TILE] add preshuffleB mode for ABQuant GEMM (#3495)

2026-01-06 12:35:01 -08:00

test_gemm_quant_abquant_preshuffleQuant.cpp

[CK_Tile] Adding support for preshuffleQuant in AB quant Block Scale Gemm (#3629)

2026-01-28 19:45:09 -08:00

test_gemm_quant_aquant_base_ccr.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_aquant_base_rcr.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_aquant_base_rrr_crr.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_aquant_mem_decode_interwave.cpp

feat: Add Interwave scheduler for aquant memory pipeline (#3540)

2026-01-26 11:27:42 -08:00

test_gemm_quant_aquant_mem_decode_intrawave.cpp

feat: Add Interwave scheduler for aquant memory pipeline (#3540)

2026-01-26 11:27:42 -08:00

test_gemm_quant_aquant_mem_prefill_interwave.cpp

feat: Add Interwave scheduler for aquant memory pipeline (#3540)

2026-01-26 11:27:42 -08:00

test_gemm_quant_aquant_prefill.cpp

feat: Add Interwave scheduler for aquant memory pipeline (#3540)

2026-01-26 11:27:42 -08:00

test_gemm_quant_aquant_preshuffle.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_aquant_transpose_c.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_base.hpp

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

test_gemm_quant_bquant_1d_64.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_1d_128.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_2d_large_n.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_2d_medium_n.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_2d_small_n.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_preshuffle_decode_1d.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_preshuffle_decode_2d.cpp

[CK_Tile] Support for group size 128 for Preshuffle quant for 2d block scale gemm (#3462)

2026-01-14 10:00:19 -08:00

test_gemm_quant_bquant_preshuffle_prefill_1d.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_preshuffle_prefill_2d.cpp

[CK_Tile] Support for group size 128 for Preshuffle quant for 2d block scale gemm (#3462)

2026-01-14 10:00:19 -08:00

test_gemm_quant_bquant_preshuffle_tiled_permute.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_bquant_preshuffleQuant_decode_1d.cpp

[CK_Tile] Support for group size 128 for Preshuffle quant for 2d block scale gemm (#3462)

2026-01-14 10:00:19 -08:00

test_gemm_quant_bquant_preshuffleQuant_decode_2d.cpp

[CK_Tile] Support for group size 128 for Preshuffle quant for 2d block scale gemm (#3462)

2026-01-14 10:00:19 -08:00

test_gemm_quant_bquant_preshuffleQuant_prefill_1d.cpp

[CK_Tile] Support for group size 128 for Preshuffle quant for 2d block scale gemm (#3462)

2026-01-14 10:00:19 -08:00

test_gemm_quant_bquant_preshuffleQuant_prefill_2d.cpp

[CK_Tile] Support for group size 128 for Preshuffle quant for 2d block scale gemm (#3462)

2026-01-14 10:00:19 -08:00

test_gemm_quant_bquant_splitk_decode.cpp

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

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

test_gemm_quant_bquant_splitk_prefill.cpp

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

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

test_gemm_quant_bquant_transpose.cpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395)

2025-12-15 07:19:29 -08:00

test_gemm_quant_fixtures.hpp

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

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

test_gemm_quant_rowcol.cpp

chore: update copyright header for misc files (#3402)

2025-12-11 08:25:29 -08:00

test_gemm_quant_tensor.cpp

chore: update copyright header for misc files (#3402)

2025-12-11 08:25:29 -08:00