Check Torch minver/maxver handling with Nix

Author: danieldk

.github/workflows/nix_fmt.yaml

@@ -1,4 +1,4 @@
-name: "Check Nix formatting"
+name: "Nix checks"
 on:
   push:
     branches: [main]
@@ -9,7 +9,7 @@ on:
 
 jobs:
   build:
-    name: Check Nix formatting
+    name: Nix checks
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -18,3 +18,5 @@ jobs:
           nix_path: nixpkgs=channel:nixos-unstable
       - name: Check formatting
         run: nix fmt -- --ci
+      - name: Nix checks
+        run: nix build .\#checks.x86_64-linux.default

examples/relu-torch-bounds/build.toml

@@ -4,7 +4,7 @@ universal = false
 
 [torch]
 src = ["torch-ext/torch_binding.cpp", "torch-ext/torch_binding.h"]
-minver = "2.8"
+minver = "2.9"
 maxver = "2.9"
 
 [kernel.relu]

flake.nix

@@ -118,6 +118,11 @@
 
       in
       rec {
+        checks.default = pkgs.callPackage ./lib/checks.nix {
+          inherit buildSets;
+          build = defaultBuildPerSystem.${system};
+        };
+
         formatter = pkgs.nixfmt-tree;
 
         packages =

lib/checks.nix

@@ -0,0 +1,20 @@
+{
+  lib,
+  runCommand,
+
+  build,
+  buildSets,
+}:
+
+let
+  kernelBuildSets = build.applicableBuildSets {
+    inherit buildSets;
+    path = ../examples/relu-torch-bounds;
+  };
+in
+assert lib.assertMsg (builtins.all (buildSet: buildSet.torch.version == "2.9.0") kernelBuildSets) ''
+  Torch minver/maxver filtering does not work.
+'';
+runCommand "builder-nix-checks" { } ''
+  touch $out
+''