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