Build uploaded using `kernels`.

build/torch-cuda/_ops.py

@@ -1,8 +1,8 @@
 import torch
-ops = torch.ops._scattermoe_faabc95
+ops = torch.ops._scattermoe_441f3e5
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_scattermoe_faabc95::{op_name}"
+    return f"_scattermoe_441f3e5::{op_name}"

build/torch-cuda/metadata.json

@@ -1,4 +1,8 @@
 {
   "version": 1,
-  "python-depends": []
+  "license": "Apache-2.0",
+  "python-depends": [],
+  "backend": {
+    "type": "cuda"
+  }
 }

build/torch-cuda/scattermoe/__init__.py

@@ -1,10 +1,10 @@
 import ctypes
+import importlib.util
 import sys
-
-import importlib
 from pathlib import Path
 from types import ModuleType
 
+
 def _import_from_path(file_path: Path) -> ModuleType:
     # We cannot use the module name as-is, after adding it to `sys.modules`,
     # it would also be used for other imports. So, we make a module name that