Apoorva Kalyani 455275de80 temp work saved, changed the BDataType to f16 or bf16 since wmma currently not support non-equal A and B datatype ...

(cherry picked from commit 22fbd68f1db458ab50780a394ee2544c7a1484d1)

Co-authored-by: Cenxuan <cenxuan@streamhpc.com>

2025-06-19 09:47:06 +00:00

..

block

WMMA GEMM universal pipeline v1, mixed precision and paddings, examples (#2230)

2025-06-04 12:22:33 +06:00

device

Add a special wrapper to use DeviceGemmMultipleD_Wmma_CShuffleV3 with old API

2025-06-04 12:34:26 +05:00

element

Add Clamp/Relu bf16/fp16 cast fixes (#2279)

2025-06-03 18:31:46 +02:00

grid

Implement DeviceGemmMultipleD_Wmma_CShuffleV3

2025-06-04 12:29:34 +05:00

thread

Moe gemm activation (#2026)

2025-04-23 10:35:34 +08:00

warp

temp work saved, changed the BDataType to f16 or bf16 since wmma currently not support non-equal A and B datatype

2025-06-19 09:47:06 +00:00