be86063
35815ed
be86063
35815ed |
|
import torch
from . import _flash_attn2_870e782_dirty
ops = torch.ops._flash_attn2_870e782_dirty
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_attn2_870e782_dirty::{op_name}" |
