danieldk's picture
danieldk HF Staff
Build (x86_64-linux Torch 2.8)
d17b8f3
raw
history blame contribute delete
248 Bytes
import torch
from . import _paged_attention_8b32f11_dirty
ops = torch.ops._paged_attention_8b32f11_dirty
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_paged_attention_8b32f11_dirty::{op_name}"