diff --git a/hardware/amd/rocm.nix b/hardware/amd/rocm.nix index 11a78f7..710b896 100644 --- a/hardware/amd/rocm.nix +++ b/hardware/amd/rocm.nix @@ -114,6 +114,10 @@ in { gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}"); }; }); + + rocblas = prev.rocmPackages.rocblas.override { + gpuTargets = lib.forEach localCfg.gpuTargets (target: "gfx${target}"); + }; }; })]; };