colin
/
nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rocmPackages: fix rocmlir build (e2e test generation)

This commit is contained in:
Martin Schwaighofer 2024-03-02 18:39:14 +01:00
parent 68e19d4242
commit 1c842526a4
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/rocm-modules/6/default.nix
View File

@ -214,7 +214,7 @@ in rec {
};
rocmlir = callPackage ./rocmlir {
inherit rocmUpdateScript rocm-cmake clr;
inherit rocmUpdateScript rocm-cmake rocminfo clr;
stdenv = llvm.rocmClangStdenv;
};

7
pkgs/development/rocm-modules/6/rocmlir/default.nix
View File

@ -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;