gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rocmPackages.rocblas: move tests and benchmarks to passthru.tests

Browse Source 
Test and bench data bloated rocblas' closure size by around 1.1G
This commit is contained in:
Luna Nova
2025-10-20 20:47:05 -07:00
parent eb91372592
commit e96455db68
2 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pkgs/development/rocm-modules/6/default.nix
View File

@@ -134,10 +134,7 @@ let
; ;
}; };
rocblas = self.callPackage ./rocblas { rocblas = self.callPackage ./rocblas { };
buildTests = true;
buildBenchmarks = true;
};
rocsolver = self.callPackage ./rocsolver { }; rocsolver = self.callPackage ./rocsolver { };

21
pkgs/development/rocm-modules/6/rocblas/default.nix
View File

@@ -25,8 +25,8 @@
rocm-smi, rocm-smi,
pkg-config, pkg-config,
buildTensile ? true, buildTensile ? true,
buildTests ? true, buildTests ? false,
buildBenchmarks ? true, buildBenchmarks ? false,
tensileSepArch ? true, tensileSepArch ? true,
tensileLazyLib ? true, tensileLazyLib ? true,
withHipBlasLt ? true, withHipBlasLt ? true,
@@ -130,8 +130,6 @@ stdenv.mkDerivation (finalAttrs: {
(lib.cmakeBool "Tensile_LAZY_LIBRARY_LOADING" tensileLazyLib) (lib.cmakeBool "Tensile_LAZY_LIBRARY_LOADING" tensileLazyLib)
]; ];
passthru.amdgpu_targets = gpuTargets';
patches = [ patches = [
(fetchpatch { (fetchpatch {
name = "Extend-rocBLAS-HIP-ISA-compatibility.patch"; name = "Extend-rocBLAS-HIP-ISA-compatibility.patch";
@@ -150,10 +148,17 @@ stdenv.mkDerivation (finalAttrs: {
--replace-fail '0.10' '1.0' --replace-fail '0.10' '1.0'
''; '';
passthru.updateScript = rocmUpdateScript { passthru = {
name = finalAttrs.pname; amdgpu_targets = gpuTargets';
inherit (finalAttrs.src) owner; tests.rocblas-tests = finalAttrs.finalPackage.override {
inherit (finalAttrs.src) repo; buildBenchmarks = true;
buildTests = true;
};
updateScript = rocmUpdateScript {
name = finalAttrs.pname;
inherit (finalAttrs.src) owner;
inherit (finalAttrs.src) repo;
};
}; };
enableParallelBuilding = true; enableParallelBuilding = true;