Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.7 kB · 84 lines
Nix
at litex
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485{ buildPythonPackage, addOpenGLRunpath, fetchPypi, fetchFromGitHub, mako, boost, numpy, pytools, pytest, decorator, appdirs, six, cudaPackages, python, mkDerivation, lib}:let compyte = import ./compyte.nix { inherit mkDerivation fetchFromGitHub; };
inherit (cudaPackages) cudatoolkit;inbuildPythonPackage rec { pname = "pycuda"; version = "2022.2.2";
src = fetchPypi { inherit pname version; hash = "sha256-zZLnJGu0WsNFKVWhEHFBEmdM3ztKni9P8lpBWcaE5rs="; };
preConfigure = with lib.versions; '' ${python.pythonForBuild.interpreter} configure.py --boost-inc-dir=${boost.dev}/include \ --boost-lib-dir=${boost}/lib \ --no-use-shipped-boost \ --boost-python-libname=boost_python${major python.version}${minor python.version} \ --cuda-root=${cudatoolkit} '';
postInstall = '' ln -s ${compyte} $out/${python.sitePackages}/pycuda/compyte '';
postFixup = '' find $out/lib -type f \( -name '*.so' -or -name '*.so.*' \) | while read lib; do echo "setting opengl runpath for $lib..." addOpenGLRunpath "$lib" done '';
# Requires access to libcuda.so.1 which is provided by the driver doCheck = false;
checkPhase = '' py.test '';
nativeBuildInputs = [ addOpenGLRunpath ];
propagatedBuildInputs = [ numpy pytools pytest decorator appdirs six cudatoolkit compyte python mako ];
meta = with lib; { homepage = "https://github.com/inducer/pycuda/"; description = "CUDA integration for Python."; license = licenses.mit; maintainers = with maintainers; [ artuuge ]; };
}