danieldk's picture
danieldk HF Staff
Build (aarch64)
a9bb8f7
raw
history blame
230 Bytes
import torch
from . import _paged_attention_dde9676
ops = torch.ops._paged_attention_dde9676
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_paged_attention_dde9676::{op_name}"