tree-wide: incorporate common out-of-tree cudaSupport overlays

This commit is contained in:
Someone Serge 2023-03-31 15:51:32 +03:00
parent 1afd4dbfe5
commit 4fafb3b90b
No known key found for this signature in database
GPG key ID: 7B0E3B1390D61DA4
5 changed files with 9 additions and 6 deletions

View file

@ -129,7 +129,7 @@ backendStdenv.mkDerivation rec {
(lib.getLib libtiff)
qt6Packages.qtwayland
rdma-core
ucx
(ucx.override { enableCuda = false; }) # Avoid infinite recursion
xorg.libxshmfence
xorg.libxkbfile
] ++ (lib.optionals (lib.versionAtLeast version "12.1") (map lib.getLib ([

View file

@ -1,9 +1,9 @@
{ lib, stdenv, fetchurl, gfortran, perl, libnl
, rdma-core, zlib, numactl, libevent, hwloc, targetPackages, symlinkJoin
, libpsm2, libfabric, pmix, ucx
, config
# Enable CUDA support
, cudaSupport ? false, cudatoolkit
, cudaSupport ? config.cudaSupport or false, cudatoolkit
# Enable the Sun Grid Engine bindings
, enableSGE ? false

View file

@ -1,5 +1,6 @@
{ lib, stdenv, fetchurl, gfortran, blas, lapack
, enableCuda ? false, cudatoolkit
, config
, enableCuda ? config.cudaSupport or false, cudatoolkit
}:
let

View file

@ -6,7 +6,8 @@
, fixDarwinDylibNames
, gmp
, mpfr
, enableCuda ? false
, config
, enableCuda ? config.cudaSupport or false
, cudatoolkit
}:

View file

@ -1,6 +1,7 @@
{ lib, stdenv, fetchFromGitHub, autoreconfHook, doxygen, numactl
, rdma-core, libbfd, libiberty, perl, zlib, symlinkJoin, pkg-config
, enableCuda ? false
, config
, enableCuda ? config.cudaSupport or false
, cudatoolkit
, enableRocm ? false
, rocm-core, rocm-runtime, rocm-device-libs, hip