Merge pull request #233179 from Tungsten842/composable_kernel

composable_kernel: build on big parallel
This commit is contained in:
Martin Weinelt 2023-05-21 14:53:35 +02:00 committed by GitHub
commit f39579a9b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -76,6 +76,9 @@ let
passthru.updateScript = unstableGitUpdater { }; passthru.updateScript = unstableGitUpdater { };
# Times out otherwise
requiredSystemFeatures = [ "big-parallel" ];
meta = with lib; { meta = with lib; {
description = "Performance portable programming model for machine learning tensor operators"; description = "Performance portable programming model for machine learning tensor operators";
homepage = "https://github.com/ROCmSoftwarePlatform/composable_kernel"; homepage = "https://github.com/ROCmSoftwarePlatform/composable_kernel";
@ -89,7 +92,7 @@ let
cp -a ${ck}/bin/ckProfiler $out cp -a ${ck}/bin/ckProfiler $out
''; '';
in stdenv.mkDerivation { in stdenv.mkDerivation {
inherit (ck) pname version outputs src passthru meta; inherit (ck) pname version outputs src passthru requiredSystemFeatures meta;
dontUnpack = true; dontUnpack = true;
dontPatch = true; dontPatch = true;