liangsu9988's picture
Add validated torch-cuda artifact
57baae3 verified
Raw
History Blame
183 Bytes
import torch
_OPS_NAME = '_fa4_cute_runtime_cuda_8aeea60'
ops = getattr(torch.ops, _OPS_NAME)
def add_op_namespace_prefix(op_name: str) -> str:
return f'{_OPS_NAME}::{op_name}'