8e800cedaf
cudaPackages.cuda_compat: ignore missing libs provided at runtime cudaPackages.gpus: Jetson should never build by default cudaPackages.flags: don't build Jetson capabilities by default cudaPackages: re-introduce filter for pre-existing CUDA redist packages in overrides cudaPackages: only recurseIntoAttrs for the latest of each major version cudaPackages.nvccCompatabilities: use GCC 10 through CUDA 11.5 to avoid a GLIBC incompatability cudaPackages.cutensor: acquire libcublas through cudatoolkit prior to 11.4 cudaPackages.cuda_compat: mark as broken on aarch64-linux if not targeting Jetson cudaPackages.cutensor_1_4: fix build cudaPackages: adjust use of autoPatchelfIgnoreMissingDeps cudaPackages.cuda_nvprof: remove unecessary override to add addOpenGLRunpath cudaPackages: use getExe' to avoid patchelf warning about missing meta.mainProgram cudaPackages: fix evaluation with Nix 2.3 cudaPackages: fix platform detection for Jetson/non-Jetson aarch64-linux python3Packages.tensorrt: mark as broken if required packages are missing Note: evaluating the name of the derivation will fail if tensorrt is not present, which is why we wrap the value in `lib.optionalString`. cudaPackages.flags.getNixSystem: add guard based on jetsonTargets cudaPackages.cudnn: use explicit path to patchelf cudaPackages.tensorrt: use explicit path to patchelf
57 lines
1.3 KiB
Nix
57 lines
1.3 KiB
Nix
{
|
|
cmake,
|
|
cudaPackages,
|
|
lib,
|
|
}:
|
|
let
|
|
inherit (cudaPackages)
|
|
autoAddOpenGLRunpathHook
|
|
backendStdenv
|
|
cuda_cccl
|
|
cuda_cudart
|
|
cuda_nvcc
|
|
cudatoolkit
|
|
cudaVersion
|
|
flags
|
|
libcublas
|
|
;
|
|
in
|
|
backendStdenv.mkDerivation {
|
|
pname = "saxpy";
|
|
version = "unstable-2023-07-11";
|
|
|
|
src = ./.;
|
|
|
|
strictDeps = true;
|
|
|
|
nativeBuildInputs =
|
|
[
|
|
cmake
|
|
autoAddOpenGLRunpathHook
|
|
]
|
|
++ lib.optionals (lib.versionOlder cudaVersion "11.4") [cudatoolkit]
|
|
++ lib.optionals (lib.versionAtLeast cudaVersion "11.4") [cuda_nvcc];
|
|
|
|
buildInputs =
|
|
lib.optionals (lib.versionOlder cudaVersion "11.4") [cudatoolkit]
|
|
++ lib.optionals (lib.versionAtLeast cudaVersion "11.4") [
|
|
libcublas
|
|
cuda_cudart
|
|
]
|
|
++ lib.optionals (lib.versionAtLeast cudaVersion "12.0") [cuda_cccl];
|
|
|
|
cmakeFlags = [
|
|
(lib.cmakeBool "CMAKE_VERBOSE_MAKEFILE" true)
|
|
(lib.cmakeFeature "CMAKE_CUDA_ARCHITECTURES" (
|
|
with flags; lib.concatStringsSep ";" (lib.lists.map dropDot cudaCapabilities)
|
|
))
|
|
];
|
|
|
|
meta = {
|
|
description = "A simple (Single-precision AX Plus Y) FindCUDAToolkit.cmake example for testing cross-compilation";
|
|
license = lib.licenses.mit;
|
|
maintainers = lib.teams.cuda.members;
|
|
platforms = lib.platforms.unix;
|
|
};
|
|
}
|