rocmPackages: set GPU_TARGETS for migraphx

2024-02-11 22:53:36 +01:00 · 2024-02-11 22:53:36 +01:00 · 3bc74475b1
parent 561cac4480
commit 3bc74475b1
1 changed files with 2 additions and 0 deletions
--- a/pkgs/development/rocm-modules/6/migraphx/default.nix
+++ b/pkgs/development/rocm-modules/6/migraphx/default.nix
@ -29,6 +29,7 @@
 , python3Packages
 , buildDocs ? false
 , buildTests ? false
+, gpuTargets ? clr.gpuTargets
 }:

 let
@ -109,6 +110,7 @@ in stdenv.mkDerivation (finalAttrs: {
    "-DCMAKE_INSTALL_BINDIR=bin"
    "-DCMAKE_INSTALL_LIBDIR=lib"
    "-DCMAKE_INSTALL_INCLUDEDIR=include"
+    "-DGPU_TARGETS=${lib.concatStringsSep ";" gpuTargets}"
  ];

  postPatch = ''