about summary refs log tree commit diff
path: root/pkgs/development/rocm-modules
diff options
context:
space:
mode:
authorMadoura <madouura@gmail.com>2022-12-17 23:22:27 -0600
committerMadoura <madouura@gmail.com>2022-12-18 12:39:58 -0600
commit93ec556b19a9cddc7798231536c7eed470e9da64 (patch)
treed791f9be62759a2556e33cb64667aee81365e788 /pkgs/development/rocm-modules
parent187452f58e6c5b0d4f50ca155c1a4b912006d630 (diff)
rocmUpdateScript: fix llvmPackages_rocm updating
Diffstat (limited to 'pkgs/development/rocm-modules')
-rw-r--r--pkgs/development/rocm-modules/update-script/default.nix9
1 files changed, 7 insertions, 2 deletions
diff --git a/pkgs/development/rocm-modules/update-script/default.nix b/pkgs/development/rocm-modules/update-script/default.nix
index 1d2b268a34b6b..6188587de31ab 100644
--- a/pkgs/development/rocm-modules/update-script/default.nix
+++ b/pkgs/development/rocm-modules/update-script/default.nix
@@ -1,4 +1,4 @@
-{ runtimeShell
+{ lib
 , writeScript
 }:
 
@@ -10,6 +10,11 @@
 }:
 
 let
+  pname =
+    if lib.hasPrefix "rocm-llvm-" name
+    then "llvmPackages_rocm.${lib.removePrefix "rocm-llvm-" name}"
+    else name;
+
   updateScript = writeScript "update.sh" ''
     #!/usr/bin/env nix-shell
     #!nix-shell -i bash -p curl jq common-updater-scripts
@@ -22,6 +27,6 @@ let
       version="''${version}.0"
     fi
 
-    update-source-version ${name} "$version" --ignore-same-hash
+    update-source-version ${pname} "$version" --ignore-same-hash
   '';
 in [ updateScript ]