pkgs/development/libraries/arrayfire/default.nix

   1 { lib, stdenv, fetchurl, fetchFromGitHub, cmake, pkg-config
   2 , opencl-clhpp, ocl-icd, fftw, fftwFloat
   3 , blas, lapack, boost, mesa, libGLU, libGL
   4 , freeimage, python3, clfft, clblas
   5 , doxygen, buildDocs ? false
   6 , cudaSupport ? false, cudatoolkit
   7 }:
   8
   9 stdenv.mkDerivation rec {
  10   pname = "arrayfire";
  11   version = "3.6.4";
  12
  13   src = fetchurl {
  14     url = "http://arrayfire.com/arrayfire_source/arrayfire-full-${version}.tar.bz2";
  15     sha256 = "1fin7a9rliyqic3z83agkpb8zlq663q6gdxsnm156cs8s7f7rc9h";
  16   };
  17
  18   cmakeFlags = [
  19     "-DAF_BUILD_OPENCL=OFF"
  20     "-DAF_BUILD_EXAMPLES=OFF"
  21     "-DBUILD_TESTING=OFF"
  22   ] ++ lib.optional cudaSupport "-DCMAKE_LIBRARY_PATH=${cudatoolkit}/lib/stubs";
  23
  24   patches = [ ./no-download.patch ];
  25
  26   postPatch = ''
  27     mkdir -p ./build/third_party/clFFT/src
  28     cp -R --no-preserve=mode,ownership ${clfft.src}/ ./build/third_party/clFFT/src/clFFT-ext/
  29     mkdir -p ./build/third_party/clBLAS/src
  30     cp -R --no-preserve=mode,ownership ${clblas.src}/ ./build/third_party/clBLAS/src/clBLAS-ext/
  31     mkdir -p ./build/include/CL
  32     cp -R --no-preserve=mode,ownership ${opencl-clhpp}/include/CL/cl2.hpp ./build/include/CL/cl2.hpp
  33   '';
  34
  35   preBuild = lib.optionalString cudaSupport ''
  36     export CUDA_PATH="${cudatoolkit}"
  37   '';
  38
  39   nativeBuildInputs = [
  40     cmake
  41     pkg-config
  42     python3
  43   ];
  44
  45   strictDeps = true;
  46
  47   buildInputs = [
  48     opencl-clhpp fftw fftwFloat
  49     blas lapack
  50     libGLU libGL
  51     mesa freeimage
  52     boost.out boost.dev
  53   ] ++ (lib.optional stdenv.isLinux ocl-icd)
  54     ++ (lib.optional cudaSupport cudatoolkit)
  55     ++ (lib.optional buildDocs doxygen);
  56
  57   meta = with lib; {
  58     description = "A general-purpose library for parallel and massively-parallel computations";
  59     longDescription = ''
  60       A general-purpose library that simplifies the process of developing software that targets parallel and massively-parallel architectures including CPUs, GPUs, and other hardware acceleration devices.";
  61     '';
  62     license = licenses.bsd3;
  63     homepage = "https://arrayfire.com/";
  64     platforms = platforms.linux ++ platforms.darwin;
  65     maintainers = with maintainers; [ chessai ];
  66   };
  67 }