112 lines
3.6 KiB
Nix
112 lines
3.6 KiB
Nix
final: prev:
|
|
let
|
|
inherit (prev) lib pkgs;
|
|
in
|
|
(lib.filterAttrs (attr: _: (prev ? "${attr}")) {
|
|
### Overrides to fix the components of cudatoolkit-redist
|
|
|
|
# Attributes that don't exist in the previous set are removed.
|
|
# That means only overrides can go here, and not new expressions!
|
|
|
|
libcufile = prev.libcufile.overrideAttrs (oldAttrs: {
|
|
buildInputs = oldAttrs.buildInputs ++ [
|
|
prev.libcublas
|
|
pkgs.numactl
|
|
pkgs.rdma-core
|
|
];
|
|
# libcuda needs to be resolved during runtime
|
|
autoPatchelfIgnoreMissingDeps = true;
|
|
});
|
|
|
|
libcusolver = final.addBuildInputs prev.libcusolver [
|
|
prev.libcublas
|
|
];
|
|
|
|
cuda_nvcc = prev.cuda_nvcc.overrideAttrs (oldAttrs:
|
|
let
|
|
inherit (prev.backendStdenv) cc;
|
|
in
|
|
{
|
|
# Point NVCC at a compatible compiler
|
|
# FIXME: non-redist cudatoolkit copy-pastes this code
|
|
|
|
# For CMake-based projects:
|
|
# https://cmake.org/cmake/help/latest/module/FindCUDA.html#input-variables
|
|
# https://cmake.org/cmake/help/latest/envvar/CUDAHOSTCXX.html
|
|
# https://cmake.org/cmake/help/latest/variable/CMAKE_CUDA_HOST_COMPILER.html
|
|
|
|
# For non-CMake projects:
|
|
# We prepend --compiler-bindir to nvcc flags.
|
|
# Downstream packages can override these, because NVCC
|
|
# uses the last --compiler-bindir it gets on the command line.
|
|
# FIXME: this results in "incompatible redefinition" warnings.
|
|
# https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#compiler-bindir-directory-ccbin
|
|
postInstall = (oldAttrs.postInstall or "") + ''
|
|
mkdir -p $out/nix-support
|
|
cat <<EOF >> $out/nix-support/setup-hook
|
|
cmakeFlags+=' -DCUDA_HOST_COMPILER=${cc}/bin'
|
|
cmakeFlags+=' -DCMAKE_CUDA_HOST_COMPILER=${cc}/bin'
|
|
if [ -z "\''${CUDAHOSTCXX-}" ]; then
|
|
export CUDAHOSTCXX=${cc}/bin;
|
|
fi
|
|
export NVCC_PREPEND_FLAGS+=' --compiler-bindir=${cc}/bin'
|
|
EOF
|
|
'';
|
|
});
|
|
|
|
cuda_nvprof = prev.cuda_nvprof.overrideAttrs (oldAttrs: {
|
|
nativeBuildInputs = oldAttrs.nativeBuildInputs ++ [ pkgs.addOpenGLRunpath ];
|
|
buildInputs = oldAttrs.buildInputs ++ [ prev.cuda_cupti ];
|
|
# libcuda needs to be resolved during runtime
|
|
autoPatchelfIgnoreMissingDeps = true;
|
|
});
|
|
|
|
cuda_demo_suite = final.addBuildInputs prev.cuda_demo_suite [
|
|
pkgs.freeglut
|
|
pkgs.libGLU
|
|
pkgs.libglvnd
|
|
pkgs.mesa
|
|
prev.libcufft
|
|
prev.libcurand
|
|
];
|
|
|
|
nsight_compute = prev.nsight_compute.overrideAttrs (oldAttrs: {
|
|
nativeBuildInputs = oldAttrs.nativeBuildInputs
|
|
++ (if (lib.versionOlder prev.nsight_compute.version "2022.2.0")
|
|
then [ pkgs.qt5.wrapQtAppsHook ]
|
|
else [ pkgs.qt6.wrapQtAppsHook ]);
|
|
buildInputs = oldAttrs.buildInputs
|
|
++ (if (lib.versionOlder prev.nsight_compute.version "2022.2.0")
|
|
then [ pkgs.qt5.qtwebview ]
|
|
else [ pkgs.qt6.qtwebview ]);
|
|
});
|
|
|
|
nsight_systems = prev.nsight_systems.overrideAttrs (oldAttrs: {
|
|
nativeBuildInputs = oldAttrs.nativeBuildInputs ++ [
|
|
pkgs.addOpenGLRunpath
|
|
pkgs.qt5.wrapQtAppsHook
|
|
];
|
|
buildInputs = oldAttrs.buildInputs ++ [
|
|
pkgs.alsa-lib
|
|
pkgs.e2fsprogs
|
|
pkgs.nss
|
|
pkgs.numactl
|
|
pkgs.pulseaudio
|
|
pkgs.wayland
|
|
pkgs.xorg.libXcursor
|
|
pkgs.xorg.libXdamage
|
|
pkgs.xorg.libXrandr
|
|
pkgs.xorg.libXtst
|
|
];
|
|
# libcuda needs to be resolved during runtime
|
|
autoPatchelfIgnoreMissingDeps = true;
|
|
});
|
|
|
|
nvidia_driver = prev.nvidia_driver.overrideAttrs (oldAttrs: {
|
|
# libcuda needs to be resolved during runtime
|
|
autoPatchelfIgnoreMissingDeps = true;
|
|
# No need to support this package as we have drivers already
|
|
# in linuxPackages.
|
|
meta.broken = true;
|
|
});
|
|
})
|