Loading pkgs/development/cuda-modules/packages/cudnn-frontend/package.nix +7 −9 Original line number Diff line number Diff line Loading @@ -2,13 +2,18 @@ autoAddDriverRunpath, catch2_3, cmake, cudaPackages, fetchFromGitHub, gitUpdater, lib, ninja, nlohmann_json, stdenv, cuda_cccl ? null, cuda_cudart ? null, cuda_nvcc ? null, cuda_nvrtc ? null, cudnn ? null, libcublas ? null, }: let inherit (lib.lists) optionals; Loading @@ -17,14 +22,6 @@ let cmakeFeature optionalString ; inherit (cudaPackages) cuda_cccl cuda_cudart cuda_nvcc cuda_nvrtc cudnn libcublas ; in # TODO(@connorbaker): This should be a hybrid C++/Python package. Loading Loading @@ -125,6 +122,7 @@ stdenv.mkDerivation (finalAttrs: { description = "A c++ wrapper for the cudnn backend API"; homepage = "https://github.com/NVIDIA/cudnn-frontend"; license = lib.licenses.mit; badPlatforms = optionals (cudnn == null) finalAttrs.meta.platforms; platforms = [ "aarch64-linux" "x86_64-linux" Loading Loading
pkgs/development/cuda-modules/packages/cudnn-frontend/package.nix +7 −9 Original line number Diff line number Diff line Loading @@ -2,13 +2,18 @@ autoAddDriverRunpath, catch2_3, cmake, cudaPackages, fetchFromGitHub, gitUpdater, lib, ninja, nlohmann_json, stdenv, cuda_cccl ? null, cuda_cudart ? null, cuda_nvcc ? null, cuda_nvrtc ? null, cudnn ? null, libcublas ? null, }: let inherit (lib.lists) optionals; Loading @@ -17,14 +22,6 @@ let cmakeFeature optionalString ; inherit (cudaPackages) cuda_cccl cuda_cudart cuda_nvcc cuda_nvrtc cudnn libcublas ; in # TODO(@connorbaker): This should be a hybrid C++/Python package. Loading Loading @@ -125,6 +122,7 @@ stdenv.mkDerivation (finalAttrs: { description = "A c++ wrapper for the cudnn backend API"; homepage = "https://github.com/NVIDIA/cudnn-frontend"; license = lib.licenses.mit; badPlatforms = optionals (cudnn == null) finalAttrs.meta.platforms; platforms = [ "aarch64-linux" "x86_64-linux" Loading