composable_kernel: revert split output workaround
This commit is contained in:
parent
4e6190bec3
commit
601c3752a2
|
@ -13,104 +13,74 @@
|
||||||
, gpuTargets ? [ ] # gpuTargets = [ "gfx803" "gfx900" "gfx1030" ... ]
|
, gpuTargets ? [ ] # gpuTargets = [ "gfx803" "gfx900" "gfx1030" ... ]
|
||||||
}:
|
}:
|
||||||
|
|
||||||
let
|
stdenv.mkDerivation (finalAttrs: {
|
||||||
# This is now over 3GB, to allow hydra caching we separate it
|
pname = "composable_kernel";
|
||||||
ck = stdenv.mkDerivation (finalAttrs: {
|
version = "unstable-2023-01-16";
|
||||||
pname = "composable_kernel";
|
|
||||||
version = "unstable-2023-01-16";
|
|
||||||
|
|
||||||
outputs = [
|
outputs = [
|
||||||
"out"
|
"out"
|
||||||
] ++ lib.optionals buildTests [
|
] ++ lib.optionals buildTests [
|
||||||
"test"
|
"test"
|
||||||
] ++ lib.optionals buildExamples [
|
] ++ lib.optionals buildExamples [
|
||||||
"example"
|
"example"
|
||||||
];
|
];
|
||||||
|
|
||||||
# ROCm 5.6 should release composable_kernel as stable with a tag in the future
|
# ROCm 5.6 should release composable_kernel as stable with a tag in the future
|
||||||
src = fetchFromGitHub {
|
src = fetchFromGitHub {
|
||||||
owner = "ROCmSoftwarePlatform";
|
owner = "ROCmSoftwarePlatform";
|
||||||
repo = "composable_kernel";
|
repo = "composable_kernel";
|
||||||
rev = "80e05267417f948e4f7e63c0fe807106d9a0c0ef";
|
rev = "80e05267417f948e4f7e63c0fe807106d9a0c0ef";
|
||||||
hash = "sha256-+c0E2UtlG/abweLwCWWjNHDO5ZvSIVKwwwettT9mqR4=";
|
hash = "sha256-+c0E2UtlG/abweLwCWWjNHDO5ZvSIVKwwwettT9mqR4=";
|
||||||
};
|
};
|
||||||
|
|
||||||
nativeBuildInputs = [
|
nativeBuildInputs = [
|
||||||
cmake
|
cmake
|
||||||
rocm-cmake
|
rocm-cmake
|
||||||
hip
|
hip
|
||||||
clang-tools-extra
|
clang-tools-extra
|
||||||
];
|
];
|
||||||
|
|
||||||
buildInputs = [
|
buildInputs = [
|
||||||
openmp
|
openmp
|
||||||
];
|
];
|
||||||
|
|
||||||
cmakeFlags = [
|
cmakeFlags = [
|
||||||
"-DCMAKE_C_COMPILER=hipcc"
|
"-DCMAKE_C_COMPILER=hipcc"
|
||||||
"-DCMAKE_CXX_COMPILER=hipcc"
|
"-DCMAKE_CXX_COMPILER=hipcc"
|
||||||
] ++ lib.optionals (gpuTargets != [ ]) [
|
] ++ lib.optionals (gpuTargets != [ ]) [
|
||||||
"-DGPU_TARGETS=${lib.concatStringsSep ";" gpuTargets}"
|
"-DGPU_TARGETS=${lib.concatStringsSep ";" gpuTargets}"
|
||||||
"-DAMDGPU_TARGETS=${lib.concatStringsSep ";" gpuTargets}"
|
"-DAMDGPU_TARGETS=${lib.concatStringsSep ";" gpuTargets}"
|
||||||
] ++ lib.optionals buildTests [
|
] ++ lib.optionals buildTests [
|
||||||
"-DGOOGLETEST_DIR=${gtest.src}" # Custom linker names
|
"-DGOOGLETEST_DIR=${gtest.src}" # Custom linker names
|
||||||
];
|
];
|
||||||
|
|
||||||
# No flags to build selectively it seems...
|
# No flags to build selectively it seems...
|
||||||
postPatch = lib.optionalString (!buildTests) ''
|
postPatch = lib.optionalString (!buildTests) ''
|
||||||
substituteInPlace CMakeLists.txt \
|
substituteInPlace CMakeLists.txt \
|
||||||
--replace "add_subdirectory(test)" ""
|
--replace "add_subdirectory(test)" ""
|
||||||
'' + lib.optionalString (!buildExamples) ''
|
'' + lib.optionalString (!buildExamples) ''
|
||||||
substituteInPlace CMakeLists.txt \
|
substituteInPlace CMakeLists.txt \
|
||||||
--replace "add_subdirectory(example)" ""
|
--replace "add_subdirectory(example)" ""
|
||||||
'';
|
'';
|
||||||
|
|
||||||
postInstall = lib.optionalString buildTests ''
|
postInstall = lib.optionalString buildTests ''
|
||||||
mkdir -p $test/bin
|
mkdir -p $test/bin
|
||||||
mv $out/bin/test_* $test/bin
|
mv $out/bin/test_* $test/bin
|
||||||
'' + lib.optionalString buildExamples ''
|
|
||||||
mkdir -p $example/bin
|
|
||||||
mv $out/bin/example_* $example/bin
|
|
||||||
'';
|
|
||||||
|
|
||||||
passthru.updateScript = unstableGitUpdater { };
|
|
||||||
|
|
||||||
# Times out otherwise
|
|
||||||
requiredSystemFeatures = [ "big-parallel" ];
|
|
||||||
|
|
||||||
meta = with lib; {
|
|
||||||
description = "Performance portable programming model for machine learning tensor operators";
|
|
||||||
homepage = "https://github.com/ROCmSoftwarePlatform/composable_kernel";
|
|
||||||
license = with licenses; [ mit ];
|
|
||||||
maintainers = teams.rocm.members;
|
|
||||||
platforms = platforms.linux;
|
|
||||||
};
|
|
||||||
});
|
|
||||||
|
|
||||||
in stdenv.mkDerivation {
|
|
||||||
inherit (ck) pname version outputs src passthru requiredSystemFeatures meta;
|
|
||||||
|
|
||||||
dontUnpack = true;
|
|
||||||
dontPatch = true;
|
|
||||||
dontConfigure = true;
|
|
||||||
dontBuild = true;
|
|
||||||
|
|
||||||
installPhase = ''
|
|
||||||
runHook preInstall
|
|
||||||
|
|
||||||
mkdir -p $out
|
|
||||||
cp -an ${ck}/* $out
|
|
||||||
'' + lib.optionalString buildTests ''
|
|
||||||
cp -a ${ck.test} $test
|
|
||||||
'' + lib.optionalString buildExamples ''
|
'' + lib.optionalString buildExamples ''
|
||||||
cp -a ${ck.example} $example
|
mkdir -p $example/bin
|
||||||
'' + ''
|
mv $out/bin/example_* $example/bin
|
||||||
runHook postInstall
|
|
||||||
'';
|
'';
|
||||||
|
|
||||||
# Fix paths
|
passthru.updateScript = unstableGitUpdater { };
|
||||||
preFixup = ''
|
|
||||||
substituteInPlace $out/lib/cmake/composable_kernel/*.cmake \
|
# Times out otherwise
|
||||||
--replace "${ck}" "$out"
|
requiredSystemFeatures = [ "big-parallel" ];
|
||||||
'';
|
|
||||||
}
|
meta = with lib; {
|
||||||
|
description = "Performance portable programming model for machine learning tensor operators";
|
||||||
|
homepage = "https://github.com/ROCmSoftwarePlatform/composable_kernel";
|
||||||
|
license = with licenses; [ mit ];
|
||||||
|
maintainers = teams.rocm.members;
|
||||||
|
platforms = platforms.linux;
|
||||||
|
};
|
||||||
|
})
|
||||||
|
|
Loading…
Reference in a new issue