Kernels
sonic-moe / build /torch-cuda /_ops_compat.py
kernels-bot's picture
Build uploaded using `kernels`.
ac0e6e3 verified
raw
history blame contribute delete
409 Bytes
"""Compatibility helpers for op namespacing in source and built layouts."""
try:
from ._ops import add_op_namespace_prefix as _generated_add_op_namespace_prefix
except ImportError:
def _generated_add_op_namespace_prefix(name: str) -> str:
return name if "::" in name else f"sonicmoe::{name}"
def add_op_namespace_prefix(name: str) -> str:
return _generated_add_op_namespace_prefix(name)