diff --git a/pkgs/development/rocm-modules/6/default.nix b/pkgs/development/rocm-modules/6/default.nix index 551560cadee1..982a9296a253 100644 --- a/pkgs/development/rocm-modules/6/default.nix +++ b/pkgs/development/rocm-modules/6/default.nix @@ -214,7 +214,7 @@ in rec { }; rocmlir = callPackage ./rocmlir { - inherit rocmUpdateScript rocm-cmake clr; + inherit rocmUpdateScript rocm-cmake rocminfo clr; stdenv = llvm.rocmClangStdenv; }; diff --git a/pkgs/development/rocm-modules/6/rocmlir/default.nix b/pkgs/development/rocm-modules/6/rocmlir/default.nix index 78ce510e797f..d8747009a10e 100644 --- a/pkgs/development/rocm-modules/6/rocmlir/default.nix +++ b/pkgs/development/rocm-modules/6/rocmlir/default.nix @@ -4,6 +4,7 @@ , rocmUpdateScript , cmake , rocm-cmake +, rocminfo , ninja , clr , git @@ -89,8 +90,12 @@ in stdenv.mkDerivation (finalAttrs: { patchShebangs mlir patchShebangs external/llvm-project/mlir/lib/Dialect/GPU/AmdDeviceLibsIncGen.py + # remove when no longer required + substituteInPlace mlir/test/{e2e/generateE2ETest.py,fusion/e2e/generate-fusion-tests.py} \ + --replace-fail "\"/opt/rocm/bin" "\"${rocminfo}/bin" + substituteInPlace mlir/utils/performance/common/CMakeLists.txt \ - --replace "/opt/rocm" "${clr}" + --replace-fail "/opt/rocm" "${clr}" ''; dontBuild = true;