Desktop: Enable GPU nodes (#3364)

* desktop simple nix ci * make workflow manual * fixup * fix nix shell * nix fmt * fix shell
2026-10-06 11:08:12 +08:00 · 2025-11-16 22:37:02 +00:00
parent 94414adaf8
commit ebb434692a
18 changed files with 509 additions and 254 deletions
@@ -0,0 +1,58 @@
+{ pkgs, inputs, ... }:
+
+let
+  extensions = [
+    "rust-src"
+    "rust-analyzer"
+    "clippy"
+    "cargo"
+    "rustc-dev"
+    "llvm-tools"
+  ];
+  toolchain = pkgs.rust-bin.nightly."2025-06-23".default.override {
+    inherit extensions;
+  };
+  cargo = pkgs.writeShellScriptBin "cargo" ''
+    #!${pkgs.lib.getExe pkgs.bash}
+
+    filtered_args=()
+    for arg in "$@"; do
+      case "$arg" in
+        +nightly|+nightly-*) ;;
+        *) filtered_args+=("$arg") ;;
+      esac
+    done
+
+    exec ${toolchain}/bin/cargo ${"\${filtered_args[@]}"}
+  '';
+  rustc_codegen_spirv =
+    (pkgs.makeRustPlatform {
+      cargo = toolchain;
+      rustc = toolchain;
+    }).buildRustPackage
+      (finalAttrs: {
+        pname = "rustc_codegen_spirv";
+        version = "0-unstable-2025-08-04";
+        src = pkgs.fetchFromGitHub {
+          owner = "Firestar99";
+          repo = "rust-gpu-new";
+          rev = "c12f216121820580731440ee79ebc7403d6ea04f";
+          hash = "sha256-rG1cZvOV0vYb1dETOzzbJ0asYdE039UZImobXZfKIno=";
+        };
+        cargoHash = "sha256-AEigcEc5wiBd3zLqWN/2HSbkfOVFneAqNvg9HsouZf4=";
+        cargoBuildFlags = [
+          "-p"
+          "rustc_codegen_spirv"
+          "--features=use-compiled-tools"
+          "--no-default-features"
+        ];
+        doCheck = false;
+      });
+in
+{
+  toolchain = toolchain;
+  env = {
+    RUST_GPU_PATH_OVERRIDE = "${cargo}/bin:${toolchain}/bin";
+    RUSTC_CODEGEN_SPIRV_PATH = "${rustc_codegen_spirv}/lib/librustc_codegen_spirv.so";
+  };
+}