69 lines
1.7 KiB
Nix
69 lines
1.7 KiB
Nix
|
{ stdenv
|
||
|
|
, fetchFromGitHub
|
||
|
|
, cmake
|
||
|
|
, gfortran
|
||
|
|
, blas
|
||
|
|
, boost
|
||
|
|
, python
|
||
|
|
, cudatoolkit
|
||
|
|
, nvidia_x11
|
||
|
|
}:
|
||
|
|
|
||
|
|
stdenv.mkDerivation rec {
|
||
|
|
name = "clblas-cuda-${version}";
|
||
|
|
version = "git-20160505";
|
||
|
|
|
||
|
|
src = fetchFromGitHub {
|
||
|
|
owner = "clMathLibraries";
|
||
|
|
repo = "clBLAS";
|
||
|
|
rev = "d20977ec4389c6b3751e318779410007c5e272f8";
|
||
|
|
sha256 = "1jna176cxznv7iz43svd6cjrbbf0fc2lrbpfpg4s08vc7xnwp0n4";
|
||
|
|
};
|
||
|
|
|
||
|
|
patches = [ ./platform.patch ];
|
||
|
|
|
||
|
|
postPatch = ''
|
||
|
|
sed -i -re 's/(set\(\s*Boost_USE_STATIC_LIBS\s+).*/\1OFF\ \)/g' src/CMakeLists.txt
|
||
|
|
'';
|
||
|
|
|
||
|
|
configurePhase = ''
|
||
|
|
findInputs ${boost} boost_dirs propagated-native-build-inputs
|
||
|
|
|
||
|
|
export BOOST_INCLUDEDIR=$(echo $boost_dirs | sed -e s/\ /\\n/g - | grep '\-dev')/include
|
||
|
|
export BOOST_LIBRARYDIR=$(echo $boost_dirs | sed -e s/\ /\\n/g - | grep -v '\-dev')/lib
|
||
|
|
|
||
|
|
mkdir -p Build
|
||
|
|
pushd Build
|
||
|
|
|
||
|
|
export LD_LIBRARY_PATH="${blas}/lib:${nvidia_x11}/lib"
|
||
|
|
|
||
|
|
cmake ../src -DCMAKE_INSTALL_PREFIX=$out \
|
||
|
|
-DCMAKE_BUILD_TYPE=Release \
|
||
|
|
-DOPENCL_ROOT=${cudatoolkit} \
|
||
|
|
'';
|
||
|
|
|
||
|
|
dontStrip = true;
|
||
|
|
|
||
|
|
buildInputs = [
|
||
|
|
cmake
|
||
|
|
gfortran
|
||
|
|
blas
|
||
|
|
python
|
||
|
|
cudatoolkit
|
||
|
|
nvidia_x11
|
||
|
|
];
|
||
|
|
|
||
|
|
meta = with stdenv.lib; {
|
||
|
|
homepage = https://github.com/clMathLibraries/clBLAS;
|
||
|
|
description = "A software library containing BLAS functions written in OpenCL";
|
||
|
|
longDescription = ''
|
||
|
|
This package contains a library of BLAS functions on top of OpenCL.
|
||
|
|
The current version is linked to the NVIDIA OpenCL implementation provided by the CUDA toolkit.
|
||
|
|
'';
|
||
|
|
license = licenses.asl20;
|
||
|
|
maintainers = with maintainers; [ artuuge ];
|
||
|
|
platforms = platforms.linux;
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|