drbh
feat: add inital build output
8acf152
raw
history blame
230 Bytes
import torch
from . import _flash_mla_iyfleyrgf34la
ops = torch.ops._flash_mla_iyfleyrgf34la
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_mla_iyfleyrgf34la::{op_name}"