MirrorHub/nixpkgs
0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-16 14:54:29 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge pull request #318926 from mschwaig/compress-composable-kernel-extra-derivation

Browse source 
rocmPackages.composable_kernel: compress `composable_kernel` using extra derivation
This commit is contained in:
Doron Behar 2024-07-12 17:22:51 +03:00 committed by GitHub
commit 50207701ed
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 27 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pkgs/development/rocm-modules/6/composable_kernel/default.nix
View file

@ -9,6 +9,7 @@
, clang-tools-extra
, git
, gtest
, zstd
, buildTests ? false
, buildExamples ? false
, gpuTargets ? [ ] # gpuTargets = [ "gfx803" "gfx900" "gfx1030" ... ]
@ -39,6 +40,7 @@ stdenv.mkDerivation (finalAttrs: {
rocm-cmake
clr
clang-tools-extra
zstd
];
buildInputs = [ openmp ];
@ -66,7 +68,9 @@ stdenv.mkDerivation (finalAttrs: {
''
;
postInstall = lib.optionalString buildTests ''
postInstall = ''
zstd --rm $out/lib/libdevice_operations.a
'' + lib.optionalString buildTests ''
mkdir -p $test/bin
mv $out/bin/test_* $test/bin
'' + lib.optionalString buildExamples ''

16
pkgs/development/rocm-modules/6/composable_kernel/unpack.nix Normal file
View file

@ -0,0 +1,16 @@
{ runCommandLocal,
composable_kernel_build,
zstd
}:
let
ck = composable_kernel_build;
in runCommandLocal "unpack-${ck.name}" {
nativeBuildInputs = [ zstd ];
meta = ck.meta;
} ''
mkdir -p $out
cp -r --no-preserve=mode ${ck}/* $out
zstd -dv --rm $out/lib/libdevice_operations.a.zst -o $out/lib/libdevice_operations.a
substituteInPlace $out/lib/cmake/composable_kernel/*.cmake \
--replace "${ck}" "$out"
''

10
pkgs/development/rocm-modules/6/default.nix
View file

@ -242,10 +242,12 @@ in rec {
It is still available for some time as part of rocmPackages_5.
''; # Added 2024-3-3
composable_kernel = callPackage ./composable_kernel {
inherit rocmUpdateScript rocm-cmake clr;
inherit (llvm) openmp clang-tools-extra;
stdenv = llvm.rocmClangStdenv;
composable_kernel = callPackage ./composable_kernel/unpack.nix {
composable_kernel_build = callPackage ./composable_kernel {
inherit rocmUpdateScript rocm-cmake clr;
inherit (llvm) openmp clang-tools-extra;
stdenv = llvm.rocmClangStdenv;
};
};
half = callPackage ./half {