From 8ea3d589392f848249faeba22f00eb2fa26eb02d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aleix=20Bon=C3=A9?= <aleix.boneribo@bsc.es>
Date: Mon, 16 Mar 2026 11:32:32 +0100
Subject: [PATCH] Add cudatoolkit to PATH

---
 pkgs/llvm-intel/default.nix | 3 ++-
 pkgs/llvm-intel/wrapped.nix | 7 ++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/pkgs/llvm-intel/default.nix b/pkgs/llvm-intel/default.nix
index 47cf5c2b..62e421e4 100644
--- a/pkgs/llvm-intel/default.nix
+++ b/pkgs/llvm-intel/default.nix
@@ -19,7 +19,7 @@
   symlinkJoin,
   zlib,
 
-  cudaPackages,
+  cudaPackages ? null,
   enableCuda ? true,
 }:
 
@@ -228,6 +228,7 @@ stdenv.mkDerivation {
 
   passthru = {
     inherit llvmMajorVersion;
+    inherit enableCuda cudaPackages;
   };
 
   meta = {
diff --git a/pkgs/llvm-intel/wrapped.nix b/pkgs/llvm-intel/wrapped.nix
index 026a8076..3ad2e0c3 100644
--- a/pkgs/llvm-intel/wrapped.nix
+++ b/pkgs/llvm-intel/wrapped.nix
@@ -1,5 +1,6 @@
 {
   stdenv,
+  lib,
   wrapCCWith,
   llvm-intel-unwrapped,
   gcc,
@@ -11,6 +12,10 @@ let
     cc:
     let
       targetConfig = stdenv.targetPlatform.config;
+
+      extraPath = builtins.concatStringsSep ":" (
+        [ "${gcc.cc}/bin" ] ++ lib.optionals cc.enableCuda [ "${cc.cudaPackages.cudatoolkit}/bin" ]
+      );
     in
     wrapCCWith {
       inherit cc;
@@ -40,7 +45,7 @@ let
       extraBuildCommands = ''
         # FIXME: We should find a better way to modify the PATH instead of using
         # this ugly hack. See https://jungle.bsc.es/git/rarias/bscpkgs/issues/9
-        echo 'path_backup="${gcc.cc}/bin:$path_backup"' >>$out/nix-support/cc-wrapper-hook
+        echo 'path_backup="${extraPath}:$path_backup"' >>$out/nix-support/cc-wrapper-hook
 
         # Disable hardening by default
         echo "" > $out/nix-support/add-hardening.sh