kernel
drbh
feat: pass vars into fwd and include build
39b4aba
raw
history blame
233 Bytes
import torch
from . import _flash_attn_a7165c8_dirty
ops = torch.ops._flash_attn_a7165c8_dirty
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_attn_a7165c8_dirty::{op_name}"