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