Merge pull request #273048 from SomeoneSerge/fix/cudatoolkit-hook-propagation
cudaPackages.cudatoolkit: fix the setupCudaHook not being propagated correctly
This commit is contained in:
@@ -80,7 +80,7 @@ backendStdenv.mkDerivation rec {
|
||||
]
|
||||
++ lib.optionals (lib.versionOlder version "11") [libsForQt5.wrapQtAppsHook]
|
||||
++ lib.optionals (lib.versionAtLeast version "11.8") [qt6Packages.wrapQtAppsHook];
|
||||
depsTargetTargetPropagated = [setupCudaHook];
|
||||
propagatedBuildInputs = [setupCudaHook];
|
||||
buildInputs =
|
||||
lib.optionals (lib.versionOlder version "11") [
|
||||
libsForQt5.qt5.qtwebengine
|
||||
|
||||
@@ -8,7 +8,7 @@
|
||||
, mpfr
|
||||
, config
|
||||
, enableCuda ? config.cudaSupport
|
||||
, cudatoolkit
|
||||
, cudaPackages
|
||||
}:
|
||||
|
||||
stdenv.mkDerivation rec {
|
||||
@@ -25,7 +25,11 @@ stdenv.mkDerivation rec {
|
||||
};
|
||||
|
||||
nativeBuildInputs = [
|
||||
] ++ lib.optional stdenv.isDarwin fixDarwinDylibNames;
|
||||
] ++ lib.optionals stdenv.isDarwin [
|
||||
fixDarwinDylibNames
|
||||
] ++ lib.optionals enableCuda [
|
||||
cudaPackages.cuda_nvcc
|
||||
];
|
||||
|
||||
# Use compatible indexing for lapack and blas used
|
||||
buildInputs = assert (blas.isILP64 == lapack.isILP64); [
|
||||
@@ -34,7 +38,12 @@ stdenv.mkDerivation rec {
|
||||
gfortran.cc.lib
|
||||
gmp
|
||||
mpfr
|
||||
] ++ lib.optional enableCuda cudatoolkit;
|
||||
] ++ lib.optionals enableCuda [
|
||||
cudaPackages.cuda_cudart.dev
|
||||
cudaPackages.cuda_cudart.lib
|
||||
cudaPackages.libcublas.dev
|
||||
cudaPackages.libcublas.lib
|
||||
];
|
||||
|
||||
preConfigure = ''
|
||||
# Mongoose and GraphBLAS are packaged separately
|
||||
@@ -49,9 +58,9 @@ stdenv.mkDerivation rec {
|
||||
] ++ lib.optionals blas.isILP64 [
|
||||
"CFLAGS=-DBLAS64"
|
||||
] ++ lib.optionals enableCuda [
|
||||
"CUDA_PATH=${cudatoolkit}"
|
||||
"CUDART_LIB=${cudatoolkit.lib}/lib/libcudart.so"
|
||||
"CUBLAS_LIB=${cudatoolkit}/lib/libcublas.so"
|
||||
"CUDA_PATH=${cudaPackages.cuda_nvcc}"
|
||||
"CUDART_LIB=${cudaPackages.cuda_cudart.lib}/lib/libcudart.so"
|
||||
"CUBLAS_LIB=${cudaPackages.libcublas.lib}/lib/libcublas.so"
|
||||
] ++ lib.optionals stdenv.isDarwin [
|
||||
# Unless these are set, the build will attempt to use `Accelerate` on darwin, see:
|
||||
# https://github.com/DrTimothyAldenDavis/SuiteSparse/blob/v5.13.0/SuiteSparse_config/SuiteSparse_config.mk#L368
|
||||
|
||||
Reference in New Issue
Block a user