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] override rocmPackages.clr with custom supported architectures

Browse source
This commit is contained in:
Antoine Viallon 2025-03-26 00:09:43 +01:00
parent bd7ae5346e
commit e8b457bc1c
Signed by: aviallon
GPG key ID: 186FC35EDEB25716
1 changed files with 30 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

34
hardware/amd/rocm.nix
View file

@ -47,6 +47,26 @@ in {
example = "gfx902";
type = types.string;
};
gpuTargets = mkOption {
description = "Override supported GPU ISAs in some ROCm packages.";
default = [ "803"
"900"
"906:xnack-"
"908:xnack-"
"90a:xnack+" "90a:xnack-"
"940"
"941"
"942"
"1010"
"1012"
"1030"
"1031"
"1100"
"1101"
"1102" ];
example = [ "900" "1031" ];
type = with types; nullOr (listOf string);
};
};
config = mkIf (cfg.enable && localCfg.enable) {
@ -86,9 +106,15 @@ in {
nixpkgs.config.rocmSupport = true;
nixpkgs.overlays = mkBefore [(final: prev: {
# Overlay Blender to use the HIP build if we have a compatible AMD GPU
blender = prev.blender-hip;
blender-cpu = prev.blender;
})];
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}");
};
});
};
})];
};
}