kernel
danieldk's picture
danieldk HF Staff
Build for Torch 2.8 (x86_64-linux)
f8b3277
import torch
from . import _flash_attn3_48fe103_dirty
ops = torch.ops._flash_attn3_48fe103_dirty
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_attn3_48fe103_dirty::{op_name}"