diff --git a/pkgs/development/rocm-modules/6/default.nix b/pkgs/development/rocm-modules/6/default.nix index e457eff0879f..6fed3b011105 100644 --- a/pkgs/development/rocm-modules/6/default.nix +++ b/pkgs/development/rocm-modules/6/default.nix @@ -134,10 +134,7 @@ let ; }; - rocblas = self.callPackage ./rocblas { - buildTests = true; - buildBenchmarks = true; - }; + rocblas = self.callPackage ./rocblas { }; rocsolver = self.callPackage ./rocsolver { }; diff --git a/pkgs/development/rocm-modules/6/rocblas/default.nix b/pkgs/development/rocm-modules/6/rocblas/default.nix index 33d707747056..0c9c2ee5c424 100644 --- a/pkgs/development/rocm-modules/6/rocblas/default.nix +++ b/pkgs/development/rocm-modules/6/rocblas/default.nix @@ -25,8 +25,8 @@ rocm-smi, pkg-config, buildTensile ? true, - buildTests ? true, - buildBenchmarks ? true, + buildTests ? false, + buildBenchmarks ? false, tensileSepArch ? true, tensileLazyLib ? true, withHipBlasLt ? true, @@ -130,8 +130,6 @@ stdenv.mkDerivation (finalAttrs: { (lib.cmakeBool "Tensile_LAZY_LIBRARY_LOADING" tensileLazyLib) ]; - passthru.amdgpu_targets = gpuTargets'; - patches = [ (fetchpatch { name = "Extend-rocBLAS-HIP-ISA-compatibility.patch"; @@ -150,10 +148,17 @@ stdenv.mkDerivation (finalAttrs: { --replace-fail '0.10' '1.0' ''; - passthru.updateScript = rocmUpdateScript { - name = finalAttrs.pname; - inherit (finalAttrs.src) owner; - inherit (finalAttrs.src) repo; + passthru = { + amdgpu_targets = gpuTargets'; + tests.rocblas-tests = finalAttrs.finalPackage.override { + buildBenchmarks = true; + buildTests = true; + }; + updateScript = rocmUpdateScript { + name = finalAttrs.pname; + inherit (finalAttrs.src) owner; + inherit (finalAttrs.src) repo; + }; }; enableParallelBuilding = true;