chore: use latest build image and misc

build.toml

@@ -1,5 +1,6 @@
 [general]
 name = "activation"
+universal = false
 
 [torch]
 src = [
@@ -8,7 +9,7 @@ src = [
 ]
 
 [kernel.activation]
-language = "cuda-hipify"
+backend = "rocm"
 rocm-archs = [ "gfx90a" ]
 src = [
   "activation/activation_kernels.cu",

build/torch26-cxx11-rocm62-x86_64-linux/activation/{_activation_afd2a56_dirty.abi3.so → _activation_f72121c_dirty.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:786b0de9d0e49b4b35659ed6f257a4ed620186266cff7e909a69d4eca9ced89d
+oid sha256:b358f5be0dc4f1c1d7198ca4417c74cf9626f678b89772e178154acbaee1476a
 size 2460736

build/torch26-cxx11-rocm62-x86_64-linux/activation/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _activation_afd2a56_dirty
-ops = torch.ops._activation_afd2a56_dirty
+from . import _activation_f72121c_dirty
+ops = torch.ops._activation_f72121c_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_activation_afd2a56_dirty::{op_name}"
+    return f"_activation_f72121c_dirty::{op_name}"

build/torch27-cxx11-rocm63-x86_64-linux/activation/{_activation_afd2a56_dirty.abi3.so → _activation_f72121c_dirty.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:56ecd54f469047866ceeefd9bfea7cfddd2126bc9c21ff110ba0308e337cf61d
-size 2447432
+oid sha256:d322fb12e4bd5eab4700783a6cfac4a8a9f9f21c7c61fd2ddb47253da8e182f1
+size 2447176

build/torch27-cxx11-rocm63-x86_64-linux/activation/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _activation_afd2a56_dirty
-ops = torch.ops._activation_afd2a56_dirty
+from . import _activation_f72121c_dirty
+ops = torch.ops._activation_f72121c_dirty
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_activation_afd2a56_dirty::{op_name}"
+    return f"_activation_f72121c_dirty::{op_name}"

flake.lock

@@ -2,11 +2,11 @@
   "nodes": {
     "flake-compat": {
       "locked": {
-        "lastModified": 1733328505,
-        "narHash": "sha256-NeCCThCEP3eCl2l/+27kNNK7QrwZB1IJCrXfrbv5oqU=",
+        "lastModified": 1747046372,
+        "narHash": "sha256-CIVLLkVgvHYbgI2UpXvIIBJ12HWgX+fjA8Xf8PUmqCY=",
         "owner": "edolstra",
         "repo": "flake-compat",
-        "rev": "ff81ac966bb2cae68946d5ed5fc4994f96d0ffec",
+        "rev": "9100a0f413b0c601e0533d1d94ffd501ce2e7885",
         "type": "github"
       },
       "original": {
@@ -98,18 +98,17 @@
         ]
       },
       "locked": {
-        "lastModified": 1747925434,
-        "narHash": "sha256-yjtdRMyPIFcSF1PkDwU5Rl0bmIpJ5joad5VOt/+1ZLY=",
-        "ref": "refs/heads/main",
-        "rev": "fd0376ff1fec423c91589075fb9042767558c635",
-        "shallow": true,
-        "type": "git",
-        "url": "file:///home/nixuser/kernel-builder"
+        "lastModified": 1748620233,
+        "narHash": "sha256-VULm9HgGXvo3pyfsPy3SOhoqgkuqbGSaSemvzNUbdIU=",
+        "owner": "huggingface",
+        "repo": "kernel-builder",
+        "rev": "da3340e5b3cbb6086600420f4814b033395788d1",
+        "type": "github"
       },
       "original": {
-        "shallow": true,
-        "type": "git",
-        "url": "file:///home/nixuser/kernel-builder"
+        "owner": "huggingface",
+        "repo": "kernel-builder",
+        "type": "github"
       }
     },
     "nixpkgs": {

flake.nix

@@ -1,10 +1,8 @@
 {
   description = "Flake for Torch kernel extension";
-
   inputs = {
-    kernel-builder.url = "/home/nixuser/kernel-builder";
+    kernel-builder.url = "github:huggingface/kernel-builder";
   };
-  
   outputs = { self, kernel-builder, }:
     kernel-builder.lib.genFlakeOutputs {
       path = ./.;