File size: 248 Bytes
331b7e6 |
1 2 3 4 5 6 7 8 9 |
import torch
from . import _paged_attention_xvepb4loq5mm2
ops = torch.ops._paged_attention_xvepb4loq5mm2
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_paged_attention_xvepb4loq5mm2::{op_name}" |