Added nix build

Author: dmjio

.ghci

@@ -1,4 +1,6 @@
 :set prompt "\x03BB> "
 :seti -XTypeApplications
 :load ArrayFire
+:m - ArrayFire
+import qualified ArrayFire as A
 :set -laf

arrayfire.cabal

@@ -170,9 +170,19 @@ test-suite doctests
   main-is:
     Main.hs
   hs-source-dirs:
-    doctest
+    doctests
   build-depends:
-    base, doctest >= 0.8
+     arrayfire
+   , base < 5
+   , doctest >= 0.8
+   , split
+  autogen-modules:
+    Build_doctests
+  other-modules:
+    Build_doctests
   default-language:
     Haskell2010
 
+source-repository head
+  type: git
+  location: https://github.com/arrayfire/arrayfire-haskell.git

default.nix

@@ -1,3 +1,8 @@
-{ pkgs ? import <nixpkgs> {} }:
+{ pkgs ? import <nixpkgs> { config.allowUnfree = true; } }:
 # Latest arrayfire is not yet procured w/ nix.
-pkgs.haskellPackages.callCabal2nix "arrayfire" ./. { af = null; }
+let
+  af = pkgs.callPackage ./nix {};
+  pkg = pkgs.haskellPackages.callCabal2nix "arrayfire" ./. { inherit af; };
+in
+  with pkgs.haskell.lib;
+  enableCabalFlag pkg "disable-default-paths"

doctest/Main.hs

@@ -0,0 +1,15 @@
+module Main (main) where
+
+import ArrayFire
+import Build_doctests     (flags, pkgs, module_sources)
+import System.Environment
+import Test.DocTest       (doctest)
+import Data.List.Split
+
+main :: IO ()
+main = do
+  print $ 1 + (1 :: Array Int)
+  moreFlags <- drop 1 . splitOn " " <$> getEnv "NIX_TARGET_LDFLAGS"
+  mapM_ print moreFlags
+  mapM_ print (flags ++ pkgs ++ module_sources ++ moreFlags)
+  doctest (moreFlags ++ flags ++ pkgs ++ module_sources)

doctests/Main.hs

@@ -0,0 +1,79 @@
+{ stdenv, fetchurl, fetchFromGitHub, cmake, pkgconfig
+, cudatoolkit, opencl-clhpp, ocl-icd, fftw, fftwFloat, mkl
+, blas, openblas, boost, mesa_noglu, libGLU_combined
+, freeimage, python, lib
+}:
+
+let
+  version = "3.6.4";
+
+  clfftSource = fetchFromGitHub {
+    owner = "arrayfire";
+    repo = "clFFT";
+    rev = "16925fb93338b3cac66490b5cf764953d6a5dac7";
+    sha256 = "0y35nrdz7w4n1l17myhkni3hwm37z775xn6f76xmf1ph7dbkslsc";
+    fetchSubmodules = true;
+  };
+
+  clblasSource = fetchFromGitHub {
+    owner = "arrayfire";
+    repo = "clBLAS";
+    rev = "1f3de2ae5582972f665c685b18ef0df43c1792bb";
+    sha256 = "154mz52r5hm0jrp5fqrirzzbki14c1jkacj75flplnykbl36ibjs";
+    fetchSubmodules = true;
+  };
+
+  cl2hppSource = fetchurl {
+    url = "https://github.com/KhronosGroup/OpenCL-CLHPP/releases/download/v2.0.10/cl2.hpp";
+    sha256 = "1v4q0g6b6mwwsi0kn7kbjn749j3qafb9r4ld3zdq1163ln9cwnvw";
+  };
+
+in stdenv.mkDerivation {
+  pname = "arrayfire";
+  inherit version;
+
+  src = fetchurl {
+    url = "http://arrayfire.com/arrayfire_source/arrayfire-full-${version}.tar.bz2";
+    sha256 = "1fin7a9rliyqic3z83agkpb8zlq663q6gdxsnm156cs8s7f7rc9h";
+  };
+
+  cmakeFlags = [
+    "-DAF_BUILD_OPENCL=OFF"
+    "-DAF_BUILD_EXAMPLES=OFF"
+    "-DBUILD_TESTING=OFF"
+  ] ++ (lib.optional stdenv.isLinux ["-DCMAKE_LIBRARY_PATH=${cudatoolkit}/lib/stubs"]);
+
+  patches = [ ./no-download.patch ];
+
+  postPatch = ''
+    mkdir -p ./build/third_party/clFFT/src
+    cp -R --no-preserve=mode,ownership ${clfftSource}/ ./build/third_party/clFFT/src/clFFT-ext/
+    mkdir -p ./build/third_party/clBLAS/src
+    cp -R --no-preserve=mode,ownership ${clblasSource}/ ./build/third_party/clBLAS/src/clBLAS-ext/
+    mkdir -p ./build/include/CL
+    cp -R --no-preserve=mode,ownership ${cl2hppSource} ./build/include/CL/cl2.hpp
+  '';
+
+  preBuild = lib.optionalString stdenv.isLinux ''
+    export CUDA_PATH="${cudatoolkit}"'
+  '';
+
+  enableParallelBuilding = true;
+
+  buildInputs = [
+    cmake pkgconfig
+    opencl-clhpp fftw fftwFloat
+    mkl openblas
+    libGLU_combined
+    mesa_noglu freeimage
+    boost.out boost.dev python
+    ] ++ (lib.optional stdenv.isLinux [ cudatoolkit ocl-icd ]);
+
+  meta = with stdenv.lib; {
+    description = "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";
+    license = licenses.bsd3;
+    homepage = https://arrayfire.com/ ;
+    maintainers = with stdenv.lib.maintainers; [ chessai ];
+    inherit version;
+  };
+}

nix/default.nix

@@ -0,0 +1,28 @@
+diff --git a/CMakeModules/build_clBLAS.cmake b/CMakeModules/build_clBLAS.cmake
+index 8de529e8..6361b613 100644
+--- a/CMakeModules/build_clBLAS.cmake
++++ b/CMakeModules/build_clBLAS.cmake
+@@ -14,8 +14,7 @@ find_package(OpenCL)
+ 
+ ExternalProject_Add(
+     clBLAS-ext
+-    GIT_REPOSITORY https://github.com/arrayfire/clBLAS.git
+-    GIT_TAG arrayfire-release
++    DOWNLOAD_COMMAND true
+     BUILD_BYPRODUCTS ${clBLAS_location}
+     PREFIX "${prefix}"
+     INSTALL_DIR "${prefix}"
+diff --git a/CMakeModules/build_clFFT.cmake b/CMakeModules/build_clFFT.cmake
+index 28be38a3..85e3915e 100644
+--- a/CMakeModules/build_clFFT.cmake
++++ b/CMakeModules/build_clFFT.cmake
+@@ -20,8 +20,7 @@ ENDIF()
+ 
+ ExternalProject_Add(
+     clFFT-ext
+-    GIT_REPOSITORY https://github.com/arrayfire/clFFT.git
+-    GIT_TAG arrayfire-release
++    DOWNLOAD_COMMAND true
+     PREFIX "${prefix}"
+     INSTALL_DIR "${prefix}"
+     UPDATE_COMMAND ""

nix/no-download.patch

@@ -5,6 +5,7 @@ in
   pkgs.lib.overrideDerivation pkg (drv: {
     shellHook = ''
       export PATH=$PATH:${pkgs.haskellPackages.doctest}/bin
+      export PATH=$PATH:${pkgs.haskellPackages.cabal-install}/bin
       function ghcid () {
         ${pkgs.haskellPackages.ghcid.bin}/bin/ghcid -c 'cabal v1-repl lib:arrayfire'
       };
@@ -18,13 +19,13 @@ in
          ${pkgs.ag}/bin/ag -l | \
            ${pkgs.entr}/bin/entr sh -c \
              'cabal v1-configure --enable-tests && \
-                cabal v1-build doctests && dist/build/doctests/doctests'
+                cabal v1-build doctests && dist/build/doctests/doctests src/ArrayFire/Algorithm.hs'
       }
       function repl () {
-        ${pkgs.cabal-install}/bin/cabal v1-repl lib:arrayfire
+        cabal v1-repl lib:arrayfire
       }
       function docs () {
-        ${pkgs.cabal-install}/bin/cabal haddock
+        cabal haddock
         open ./dist-newstyle/*/*/*/*/doc/html/arrayfire/index.html
       }
     '';

shell.nix

@@ -33,8 +33,11 @@ import Data.Word
 
 -- | Sum all of the elements in 'Array' along the specified dimension
 --
--- >>> sum (vector @Double 10 [1..]) 0
--- 33
+-- >>> A.sum (A.vector @Double 10 [1..]) 0
+-- 55.0
+--
+-- >>> A.sum (A.matrix @Double (10,10) [[2..],[2..]]) 0
+-- 65.0
 --
 sum
   :: AFType a
@@ -48,9 +51,9 @@ sum x (fromIntegral -> n) = getScalar (x `op1` (\p a -> af_sum p a n))
 
 -- | Sum all of the elements in 'Array' along the specified dimension, using a default value for NaN
 --
--- @
--- >>> 'sumNaN' ('vector' \@'Double' 10 [1..]) 0 0.0
--- @
+-- >>> A.sumNaN (A.vector @Double 10 [1..]) 0 0.0
+-- 55
+--
 sumNaN
   :: (Fractional a, AFType a)
   => Array a
@@ -65,9 +68,9 @@ sumNaN n (fromIntegral -> i) d = getScalar (n `op1` (\p a -> af_sum_nan p a i d)
 
 -- | Product all of the elements in 'Array' along the specified dimension
 --
--- @
--- >>> 'product' ('vector' \@'Double' 10 [1..]) 0
--- @
+-- >>> A.product (A.vector @Double 10 [1..]) 0
+-- 3628800.0
+--
 product
   :: AFType a
   => Array a
@@ -80,9 +83,9 @@ product x (fromIntegral -> n) = getScalar (x `op1` (\p a -> af_product p a n))
 
 -- | Product all of the elements in 'Array' along the specified dimension, using a default value for NaN
 --
--- @
--- >>> 'productNaN' ('vector' \@'Double' 10 [1..]) 0 0.0
--- @
+-- >>> A.productNaN (A.vector @Double 10 [1..]) 0 0.0
+-- 3628800.0
+--
 productNaN
   :: (AFType a, Fractional a)
   => Array a