ecb82b6
1
2
3
4
5
6
7
8
9
import torch from . import _mamba_ssm_18a29a2 ops = torch.ops._mamba_ssm_18a29a2 def add_op_namespace_prefix(op_name: str): """ Prefix op by namespace. """ return f"_mamba_ssm_18a29a2::{op_name}"
