From 0ae051d93a3b3278f0a68b45c44d96ebfb0df394 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 11 Nov 2025 11:56:04 +0100 Subject: [PATCH] Include CPU kernels in CI builds --- lib/gen-flake-outputs.nix | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/lib/gen-flake-outputs.nix b/lib/gen-flake-outputs.nix index 15418879..449b13c8 100644 --- a/lib/gen-flake-outputs.nix +++ b/lib/gen-flake-outputs.nix @@ -61,13 +61,15 @@ let buildConfig // { bundleBuild = buildConfig.bundleBuild or false; framework = - if buildConfig ? cudaVersion then + if buildConfig.cpu or false then + "cpu" + else if buildConfig ? cudaVersion then "cuda" else if buildConfig ? rocmVersion then "rocm" else if buildConfig ? xpuVersion then "xpu" - else if system == "aarch64-darwin" then + else if buildConfig.metal or false then "metal" else throw "Cannot determine framework for build set"; @@ -180,7 +182,8 @@ in framework: builtins.filter (set: set.buildConfig.framework == framework) buildSetsSorted; # It is too costly to build all variants in CI, so we just build one per framework. onePerFramework = - (headOrEmpty (setsWithFramework "cuda")) + (headOrEmpty (setsWithFramework "cpu")) + ++ (headOrEmpty (setsWithFramework "cuda")) ++ (headOrEmpty (setsWithFramework "metal")) ++ (headOrEmpty (setsWithFramework "rocm")) ++ (headOrEmpty (setsWithFramework "xpu"));