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

[Hardware/AMD/ROCm] attempt fix

Browse source
This commit is contained in:
Antoine Viallon 2025-05-16 10:48:41 +02:00
parent 183a38ec0b
commit 60ee53b417
Signed by: aviallon
GPG key ID: 186FC35EDEB25716
1 changed files with 15 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

29
hardware/amd/rocm.nix
View file

@ -45,7 +45,7 @@ in {
description = "What is the GFX ISA of your system. Leave blank if you have several GPUs of incompatible ISAs";
default = "";
example = "gfx902";
type = types.string;
type = types.str;
};
gpuTargets = mkOption {
description = "Override supported GPU ISAs in some ROCm packages.";
@ -105,21 +105,22 @@ in {
nixpkgs.config.rocmSupport = true;
nixpkgs.overlays = mkBefore [(final: prev: {
rocmPackages_5 = final.rocmPackages;
nixpkgs.overlays = mkIf (! isNull localCfg.gpuTargets) (mkBefore [(final: prev: {
#rocmPackages_5 = final.rocmPackages;
rocmPackages = prev.rocmPackages // {
clr = prev.rocmPackages.clr.overrideAttrs (oldAttrs: {
passthru = oldAttrs.passthru // {
# We cannot use this for each ROCm library, as each defines their own supported targets
# See: https://github.com/ROCm/ROCm/blob/77cbac4abab13046ee93d8b5bf410684caf91145/README.md#library-target-matrix
gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}");
};
});
clr = prev.rocmPackages.clr.override { localGpuTargets = localCfg.gpuTargets; };
# (oldAttrs: {
# passthru = oldAttrs.passthru // {
# # We cannot use this for each ROCm library, as each defines their own supported targets
# # See: https://github.com/ROCm/ROCm/blob/77cbac4abab13046ee93d8b5bf410684caf91145/README.md#library-target-matrix
# gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}");
# };
#});
rocblas = prev.rocmPackages.rocblas.override {
gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}");
};
#rocblas = prev.rocmPackages.rocblas.override {
# gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}");
#};
};
})];
})]);
};
}