Kernels
danieldk's picture
danieldk HF Staff
Build uploaded using `kernels`.
1731a09 verified
raw
history blame contribute delete
218 Bytes
import torch
from . import _flash_attn3_1d39a44
ops = torch.ops._flash_attn3_1d39a44
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_attn3_1d39a44::{op_name}"