Commit 81564851 authored by Markus Kowalewski's avatar Markus Kowalewski
Browse files

sirius: fix CUDA build

parent 3bd19bc4

pkgs/by-name/si/sirius/package.nix

+7 −3
Original line number Diff line number Diff line
@@ -55,7 +55,7 @@ stdenv.mkDerivation rec { 
    cmake

    gfortran

    pkg-config

  ];

  ] ++ lib.optional (gpuBackend == "cuda") cudaPackages.cuda_nvcc;



  buildInputs = [

    blas
@@ -72,8 +72,12 @@ stdenv.mkDerivation rec { 
    eigen

    libvdwxc

  ]

  ++ lib.optional (gpuBackend == "cuda") cudaPackages.cudatoolkit

  ++ lib.optionals (gpuBackend == "rocm") [

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

    cudaPackages.cuda_cudart

    cudaPackages.cuda_profiler_api

    cudaPackages.cudatoolkit

    cudaPackages.libcublas

  ] ++ lib.optionals (gpuBackend == "rocm") [

    rocmPackages.clr

    rocmPackages.rocblas

  ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp