Kernels
danieldk's picture
danieldk HF Staff
Build (aarch64-linux)
ecb82b6
raw
history blame
212 Bytes
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}"