Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.8 kB · 85 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586{ 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 = "2024.1"; format = "setuptools";
src = fetchPypi { inherit pname version; hash = "sha256-1Q0j/2NxSCz/fUuVPvQKuByd8DjsthRIT5/VNHMnMn4="; };
preConfigure = with lib.versions; '' ${python.pythonOnBuildForHost.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 ]; };
}