almersawi a547aeb828 feat(rocm-support): support mamba2 on rocm (#18565) ...

Signed-off-by: Islam Almersawi <islam.almersawi@openinnovation.ai>
Co-authored-by: Islam Almersawi <islam.almersawi@openinnovation.ai>

2025-05-27 00:07:53 -07:00

..

selective_scan_fwd.cu

feat(rocm-support): support mamba2 on rocm (#18565)

2025-05-27 00:07:53 -07:00

selective_scan.h

[Kernel][Model] Improve continuous batching for Jamba and Mamba (#9189)

2024-10-16 12:12:43 -04:00

static_switch.h

[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651)

2024-08-28 15:06:52 -07:00