aviallon/nixos-lib
1
0
Fork 0
mirror of https://github.com/aviallon/nixos-lib.git synced 2026-04-06 01:38:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Hardware/AMD/ROCm] fix tensile

Browse source
This commit is contained in:
Antoine Viallon 2025-01-29 16:40:05 +01:00
parent 8e69a7d488
commit 7da398fbff
Signed by: aviallon
GPG key ID: 186FC35EDEB25716
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
hardware/amd/rocm.nix
View file

@ -85,7 +85,7 @@ in {
nixpkgs.config.rocmSupport = true; nixpkgs.config.rocmSupport = true;
nixpkgs.overlays = [(final: prev: { nixpkgs.overlays = mkBefore [(final: prev: {
# Overlay Blender to use the HIP build if we have a compatible AMD GPU # Overlay Blender to use the HIP build if we have a compatible AMD GPU
blender = final.blender-hip; blender = final.blender-hip;
blender-cpu = prev.blender; blender-cpu = prev.blender;
@ -106,6 +106,8 @@ in {
]; ];
}); });
rocblas = prev.hello;
rocmlir-rock = rocmlir.override { rocmlir-rock = rocmlir.override {
buildRockCompiler = true; buildRockCompiler = true;
}; };
@ -113,6 +115,14 @@ in {
miopen = prev.rocmPackages.miopen.override { rocmlir = rocmlir-rock; }; miopen = prev.rocmPackages.miopen.override { rocmlir = rocmlir-rock; };
migraphx = prev.rocmPackages.migraphx.override { rocmlir = rocmlir-rock; }; migraphx = prev.rocmPackages.migraphx.override { rocmlir = rocmlir-rock; };
tensile = prev.rocmPackages.tensile.overrideAttrs (old: {
# TODO: remove this workaround once https://github.com/NixOS/nixpkgs/pull/323869
# does not cause issues anymore, or at least replace it with a better orkaround
setupHook = writeText "setup-hook" ''
export TENSILE_ROCM_ASSEMBLER_PATH="${stdenv.cc.cc}/bin/clang++";
'';
});
}; };
})]; })];
}; };