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

Merged

vllm-bot merged 8 commits intovllm-project:mainfrom

almersawi:feat/rocm-mamba2-support

May 27, 2025

Commits on May 24, 2025

feat(rocm-support): expose causal_conv1d_fwd, causal_conv1d_update and selective_scan_fwd for rocm
Islam Almersawi
committed
fix(CMake): correct CUDA supported architectures formatting
Islam Almersawi
committed
fix(causal_conv1d): correct closing brace in causal_conv1d_fwd_launch function
Islam Almersawi
committed
fix: fix precommit checks
Islam Almersawi
committed
fix(mamba2_metadata): format imports and improve error message for unsupported platforms
Islam Almersawi
committed
fix(mamba2_metadata): correct indentation in isinstance check for attn_metadata
Islam Almersawi
committed
fix(torch_bindings): remove unnecessary blank line in TORCH_LIBRARY_EXPAND definition
Islam Almersawi
committed
fix(causal_conv1d): remove unnecessary preprocessor directive in causal_conv1d_fwd_launch function
Islam Almersawi
committed