nvidia-thrust: init at 1.16.0

pkgs/development/libraries/nvidia-thrust/default.nix

@@ -0,0 +1,61 @@
+{ lib
+, fetchFromGitHub
+, stdenv
+, cmake
+, cudaPackages
+, symlinkJoin
+}:
+let
+  pname = "nvidia-thrust";
+  version = "1.16.0";
+
+  # TODO: Would like to use this:
+  cudaJoined = symlinkJoin {
+    name = "cuda-packages-unsplit";
+    paths = with cudaPackages; [
+      cuda_nvcc
+      cuda_cudart # cuda_runtime.h
+      libcublas
+    ];
+  };
+in
+stdenv.mkDerivation {
+  inherit pname version;
+
+  src = fetchFromGitHub {
+    owner = "NVIDIA";
+    repo = "thrust";
+    rev = version;
+    fetchSubmodules = true;
+    hash = "sha256-/EyznxWKuHuvHNjq+SQg27IaRbtkjXR2zlo2YgCWmUQ=";
+  };
+
+  nativeBuildInputs = [
+    cmake
+
+    # goes in native build inputs because thrust looks for headers
+    # in a path relative to nvcc...
+    #
+    # Works when build=host, but we only support
+    # cuda on x86_64 anyway
+
+    # TODO: but instead using this
+    # cudaJoined
+    cudaPackages.cudatoolkit
+  ];
+
+  cmakeFlags = [
+    "-DTHRUST_INCLUDE_CUB_CMAKE=ON"
+  ];
+
+  meta = with lib; {
+    description = ''
+      A high-level C++ parallel algorithms library
+      that builds on top of CUDA, TBB, OpenMP, etc.
+    '';
+    homepage = "https://github.com/NVIDIA/thrust";
+    license = licenses.asl20;
+    platforms = platforms.unix;
+    maintainers = with maintainers; [ SomeoneSerge ];
+  };
+}

pkgs/top-level/all-packages.nix

@@ -10306,6 +10306,8 @@ with pkgs;
 
   nvfetcher = haskell.lib.compose.justStaticExecutables haskellPackages.nvfetcher;
 
+  nvidia-thrust = callPackage ../development/libraries/nvidia-thrust { };
+
   miller = callPackage ../tools/text/miller { };
 
   milu = callPackage ../applications/misc/milu { };