import torch from . import _diffusion_step_ops_cuda_7781728 ops = torch.ops._diffusion_step_ops_cuda_7781728 def add_op_namespace_prefix(op_name: str): return f"_diffusion_step_ops_cuda_7781728::{op_name}"