mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-12 16:53:21 +00:00
nvidia-thrust: init at 1.16.0
This commit is contained in:
parent
36ba5681fa
commit
95d21285bd
61
pkgs/development/libraries/nvidia-thrust/default.nix
Normal file
61
pkgs/development/libraries/nvidia-thrust/default.nix
Normal file
@ -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 ];
|
||||
};
|
||||
}
|
@ -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 { };
|
||||
|
Loading…
Reference in New Issue
Block a user