Commit 3bd19bc4 authored by Markus Kowalewski's avatar Markus Kowalewski
Browse files

spfft: fix CUDA build

parent 702762fa

pkgs/by-name/sp/spfft/package.nix

+5 −4
Original line number Diff line number Diff line
@@ -34,13 +34,14 @@ stdenv.mkDerivation rec { 
  nativeBuildInputs = [

    cmake

    gfortran

  ];

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



  buildInputs = [

    fftw

  ]

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

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

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

    cudaPackages.libcufft

    cudaPackages.cuda_cudart

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

    rocmPackages.clr

    rocmPackages.rocfft

    rocmPackages.hipfft