68 lines
1.6 KiB
Nix
68 lines
1.6 KiB
Nix
{ lib
|
|
, stdenv
|
|
, backendStdenv
|
|
, fetchurl
|
|
, autoPatchelfHook
|
|
, autoAddOpenGLRunpathHook
|
|
, markForCudatoolkitRootHook
|
|
}:
|
|
|
|
pname:
|
|
attrs:
|
|
|
|
let
|
|
arch = "linux-x86_64";
|
|
in
|
|
backendStdenv.mkDerivation {
|
|
inherit pname;
|
|
inherit (attrs) version;
|
|
|
|
src = assert (lib.hasAttr arch attrs); fetchurl {
|
|
url = "https://developer.download.nvidia.com/compute/cuda/redist/${attrs.${arch}.relative_path}";
|
|
inherit (attrs.${arch}) sha256;
|
|
};
|
|
|
|
nativeBuildInputs = [
|
|
autoPatchelfHook
|
|
# This hook will make sure libcuda can be found
|
|
# in typically /lib/opengl-driver by adding that
|
|
# directory to the rpath of all ELF binaries.
|
|
# Check e.g. with `patchelf --print-rpath path/to/my/binary
|
|
autoAddOpenGLRunpathHook
|
|
markForCudatoolkitRootHook
|
|
];
|
|
|
|
buildInputs = [
|
|
# autoPatchelfHook will search for a libstdc++ and we're giving it
|
|
# one that is compatible with the rest of nixpkgs, even when
|
|
# nvcc forces us to use an older gcc
|
|
# NB: We don't actually know if this is the right thing to do
|
|
stdenv.cc.cc.lib
|
|
];
|
|
|
|
# Picked up by autoPatchelf
|
|
# Needed e.g. for libnvrtc to locate (dlopen) libnvrtc-builtins
|
|
appendRunpaths = [
|
|
"$ORIGIN"
|
|
];
|
|
|
|
dontBuild = true;
|
|
|
|
# TODO: choose whether to install static/dynamic libs
|
|
installPhase = ''
|
|
runHook preInstall
|
|
rm LICENSE
|
|
mkdir -p $out
|
|
mv * $out
|
|
runHook postInstall
|
|
'';
|
|
|
|
passthru.stdenv = backendStdenv;
|
|
|
|
meta = {
|
|
description = attrs.name;
|
|
license = lib.licenses.unfree;
|
|
maintainers = lib.teams.cuda.members;
|
|
platforms = lib.optionals (lib.hasAttr arch attrs) [ "x86_64-linux" ];
|
|
};
|
|
}
|