Hashem Hashemi 5a499e70d5

[Kernel][Hardware][AMD] Bf16 mfma opt for ROCm skinny GEMMs (#17071) ...

Signed-off-by: Hashem Hashemi <hashem.hashemi@amd.com>
Signed-off-by: charlifu <charlifu@amd.com>
Co-authored-by: charlifu <charlifu@amd.com>

2025-05-07 22:34:49 -07:00

..

attention

…

core

Make key optional for rotary embedding (#17566)

2025-05-07 00:11:46 -07:00

mamba

[Model] Mamba2 causal conv1d Refactor to Split Prefill and Decode Requests for Corresponding Kernels (#17146)

2025-05-06 17:59:30 -07:00

moe

…

quantization

[Kernel][Hardware][AMD] Bf16 mfma opt for ROCm skinny GEMMs (#17071)

2025-05-07 22:34:49 -07:00

__init__.py

…

allclose_default.py

…

quant_utils.py

…

test_cutlass_mla_decode.py

…

test_fused_quant_activation.py

…

test_triton_flash_attention.py

…

test_triton_unified_attention.py

[Kernel] Unified Triton kernel that doesn't distinguish between prefill + decode (#16828)

2025-05-06 18:21:48 -04:00

utils.py

…