Chih-Chieh Yang b690e34824

[Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075) ...

Signed-off-by: Chih-Chieh Yang <7364402+cyang49@users.noreply.github.com>
Signed-off-by: Chih-Chieh-Yang <7364402+cyang49@users.noreply.github.com>

2025-08-02 01:59:34 -07:00

..

attention

…

core

…

mamba

[Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075)

2025-08-02 01:59:34 -07:00

moe

…

quantization

…

__init__.py

…

allclose_default.py

…

quant_utils.py

…

test_apply_repetition_penalties.py

…

test_cutlass_mla_decode.py

…

test_flex_attention.py

…

test_fused_quant_activation.py

…

test_shuffle_rows.py

…

test_triton_flash_attention.py

…

utils.py

…