129 lines
2.2 KiB
Nix
129 lines
2.2 KiB
Nix
{ stdenv
|
|
, buildPythonPackage
|
|
, fetchFromGitHub
|
|
, cmake
|
|
, cython
|
|
, numpy
|
|
, Mako
|
|
, six
|
|
, nose
|
|
, beaker
|
|
, memcached
|
|
, pkgconfig
|
|
, glibc
|
|
, clblas
|
|
, Babel
|
|
, pygments
|
|
, scipy
|
|
, python
|
|
, cudatoolkit
|
|
, nvidia_x11
|
|
}:
|
|
buildPythonPackage rec {
|
|
name = "libgpuarray-cuda-${version}";
|
|
version = "-9998.0";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "Theano";
|
|
repo = "libgpuarray";
|
|
rev = "fc36a40526c0a8303ace6c574ffdefba7feafe17";
|
|
sha256 = "1kb0k42addqjxiahlcbv6v6271yhsmz71j12186fpy60870i7zm7";
|
|
};
|
|
|
|
doCheck = true;
|
|
|
|
configurePhase = ''
|
|
mkdir -p Build/Install
|
|
pushd Build
|
|
|
|
cmake .. -DCMAKE_BUILD_TYPE=Release \
|
|
-DCMAKE_INSTALL_PREFIX=./Install \
|
|
-DCLBLAS_ROOT_DIR=${clblas}
|
|
|
|
popd
|
|
'';
|
|
|
|
preBuild = ''
|
|
pushd Build
|
|
make
|
|
make install
|
|
|
|
function fixRunPath {
|
|
p=$(patchelf --print-rpath $1)
|
|
patchelf --set-rpath "$p:${stdenv.lib.makeLibraryPath [ cudatoolkit clblas nvidia_x11 ]}" $1
|
|
}
|
|
|
|
fixRunPath Install/lib/libgpuarray.so
|
|
|
|
popd
|
|
'';
|
|
|
|
setupPyBuildFlags = [ "-L $(pwd)/Build/Install/lib" "-I $(pwd)/Build/Install/include" ];
|
|
|
|
preInstall = ''
|
|
cp -r Build/Install $out
|
|
'';
|
|
|
|
postInstall = ''
|
|
pushd $out/${python.sitePackages}/pygpu
|
|
for f in $(find $out/pygpu -name "*.h"); do
|
|
ln -s $f $(basename $f)
|
|
done
|
|
popd
|
|
'';
|
|
checkPhase = ''
|
|
mkdir -p my_bin
|
|
pushd my_bin
|
|
|
|
cat > libgpuarray_run_tests << EOF
|
|
#!/bin/sh
|
|
if [ \$# -eq 0 ]; then
|
|
echo "No argument provided."
|
|
echo "Available tests:"
|
|
ls $out/${python.sitePackages}/pygpu/tests | grep "test_"
|
|
exit 1
|
|
else
|
|
nosetests -v "$out/${python.sitePackages}/pygpu/tests/\$@"
|
|
fi
|
|
EOF
|
|
|
|
chmod +x libgpuarray_run_tests
|
|
popd
|
|
|
|
cp -r my_bin $out/bin
|
|
'';
|
|
|
|
dontStrip = true;
|
|
|
|
propagatedBuildInputs = [
|
|
numpy
|
|
scipy
|
|
nose
|
|
six
|
|
Mako
|
|
];
|
|
|
|
buildInputs = [
|
|
cmake
|
|
cython
|
|
beaker
|
|
memcached
|
|
pkgconfig
|
|
glibc
|
|
Babel
|
|
pygments
|
|
numpy.blas
|
|
cudatoolkit
|
|
nvidia_x11
|
|
clblas
|
|
];
|
|
|
|
meta = with stdenv.lib; {
|
|
homepage = https://github.com/Theano/libgpuarray;
|
|
description = "Library to manipulate tensors on GPU.";
|
|
license = licenses.free;
|
|
maintainers = with maintainers; [ artuuge ];
|
|
platforms = platforms.linux;
|
|
};
|
|
|
|
}
|