Unverified Commit 5c806f3a authored by Pavol Rusnak's avatar Pavol Rusnak Committed by GitHub
Browse files

katago: remove cudatoolkit from deps (#386336)

parents 63045b42 abed2b7b

pkgs/by-name/ka/katago/package.nix

+18 −11
Original line number Diff line number Diff line
@@ -67,16 +67,23 @@ stdenv'.mkDerivation rec { 
      boost

    ]

    ++ lib.optionals (backend == "eigen") [ eigen ]

    ++ lib.optionals (backend == "cuda") [

      cudaPackages.cuda_cudart

      cudaPackages.cudnn

      cudaPackages.cudatoolkit

    ++ lib.optionals (backend == "cuda") (

      with cudaPackages;

      [

        cuda_cccl

        cuda_cudart

        cuda_nvcc

        cudnn

        libcublas

      ]

    ++ lib.optionals (backend == "tensorrt") [

      cudaPackages.cuda_cudart

      cudaPackages.cudatoolkit

      cudaPackages.tensorrt

    )

    ++ lib.optionals (backend == "tensorrt") (

      with cudaPackages;

      [

        cuda_cudart

        tensorrt

      ]

    )

    ++ lib.optionals (backend == "opencl") [

      opencl-headers

      ocl-icd
@@ -103,7 +110,7 @@ stdenv'.mkDerivation rec { 
      cd cpp/

    ''

    + lib.optionalString (backend == "cuda" || backend == "tensorrt") ''

      export CUDA_PATH="${cudaPackages.cudatoolkit}"

      export CUDA_PATH="${cudaPackages.cuda_nvcc}"

      export EXTRA_LDFLAGS="-L/run/opengl-driver/lib"

    '';