File size: 183 Bytes
70b4af3 | 1 2 3 4 5 6 7 8 | import torch
ops = torch.ops._flash_attn4_06b9ab8
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_attn4_06b9ab8::{op_name}" |
70b4af3 | 1 2 3 4 5 6 7 8 | import torch
ops = torch.ops._flash_attn4_06b9ab8
def add_op_namespace_prefix(op_name: str):
"""
Prefix op by namespace.
"""
return f"_flash_attn4_06b9ab8::{op_name}" |