Native AscendC Mamba2 selective scan / SSD forward-backward custom operator for Huawei Ascend 910B3 and 950PR, with CANN, torch_npu, A100 benchmarks and msprof profiling.
pytorch ssd mamba cann performance-optimization state-space-models ascend-npu selective-scan ai-infrastructure ascendc kernel-optimization mamba2 huawei-ascend torch-npu custom-operator ascend-910b triton-ascend ascend-950
-
Updated
Aug 12, 2026 - Python