Build uploaded using `kernels`.

build/torch-cuda/_ops.py

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

build/torch-cuda/metadata.json

@@ -1,3 +1,6 @@
 {
-  "python-depends": []
+  "python-depends": [],
+  "backend": {
+    "type": "cuda"
+  }
 }

build/torch-cuda/triton_kernels/__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