Update flake

flake.lock

@@ -17,11 +17,11 @@
     },
     "flake-compat_2": {
       "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": {
@@ -73,11 +73,11 @@
         "nixpkgs": "nixpkgs"
       },
       "locked": {
-        "lastModified": 1753354560,
-        "narHash": "sha256-vmOfRmr0Qm/IbZTWB2sBn+UFrABSTTA/cTg+m27Yt/E=",
+        "lastModified": 1759493343,
+        "narHash": "sha256-8fhl0gwMAnOkQbogPIVq+Fha+Yeq52FaRXfwF+F9Q+k=",
         "owner": "huggingface",
         "repo": "hf-nix",
-        "rev": "7f2aceda2a2e72cd573bdb25e5c0667fd75f89d3",
+        "rev": "b1fc3a18b52447a0f24bc6884418edc5e66082b9",
         "type": "github"
       },
       "original": {
@@ -98,33 +98,32 @@
         ]
       },
       "locked": {
-        "lastModified": 1753602110,
-        "narHash": "sha256-AEt6rSqYqSTgsKZ+2BuGezurpVC2gm+Jpjqg2D54n7E=",
+        "lastModified": 1759516823,
+        "narHash": "sha256-UJVvZHtS9c64Dm4iZRaOKWB+VHI7jzcazGH57KXWeg8=",
         "owner": "huggingface",
         "repo": "kernel-builder",
-        "rev": "2021ea0f8d9e63ada986f189d077fc301cc1c3c9",
+        "rev": "e13610a05f67b7296be9ead89ad172a0a088a1c3",
         "type": "github"
       },
       "original": {
         "owner": "huggingface",
-        "ref": "torch-2.8",
         "repo": "kernel-builder",
         "type": "github"
       }
     },
     "nixpkgs": {
       "locked": {
-        "lastModified": 1752785354,
-        "narHash": "sha256-Y33ryUz7MPqKrZwlbQcsYCUz2jAJCacRf8jbs0tYUlA=",
+        "lastModified": 1755963616,
+        "narHash": "sha256-6yD0ww/S8n+U2uPYcJZ3DRURP8Kx036GRpR2uPNZroE=",
         "owner": "nixos",
         "repo": "nixpkgs",
-        "rev": "d38025438a6ee456758dc03188ca6873a415463b",
+        "rev": "73e96df7cff5783f45e21342a75a1540c4eddce4",
         "type": "github"
       },
       "original": {
         "owner": "nixos",
+        "ref": "nixos-unstable-small",
         "repo": "nixpkgs",
-        "rev": "d38025438a6ee456758dc03188ca6873a415463b",
         "type": "github"
       }
     },

flake.nix

@@ -2,7 +2,7 @@
   description = "Flake for Hopper Flash Attention kernel";
 
   inputs = {
-    kernel-builder.url = "github:huggingface/kernel-builder/torch-2.8";
+    kernel-builder.url = "github:huggingface/kernel-builder";
   };
 
   outputs =
@@ -11,24 +11,24 @@
       kernel-builder,
     }:
     kernel-builder.lib.genFlakeOutputs {
+      inherit self;
       path = ./.;
-      rev = self.shortRev or self.dirtyShortRev or self.lastModifiedDate;
-      # Building with CDUA later than 12.4 fails with:
+      # Building with CUDA later than 12.4 fails with:
       #
       # error: 'ptxas' died due to signal 11 (Invalid memory reference)
       #
       # So, build for 12.4 only and copy to all the other build variants
       # by hand (which works fine thanks to backward compat).
-      torchVersions = [
+      torchVersions = _: [
         {
-          torchVersion = "2.8";
+          torchVersion = "2.9";
           cudaVersion = "12.4";
           cxx11Abi = true;
           systems = [
             "x86_64-linux"
             "aarch64-linux"
           ];
-          upstreamVariant = true;
+          bundleBuild = true;
         }
       ];
     };