import torch from . import _megablocks_cpu_f8f8b50 ops = torch.ops._megablocks_cpu_f8f8b50 def add_op_namespace_prefix(op_name: str): """ Prefix op by namespace. """ return f"_megablocks_cpu_f8f8b50::{op_name}"