Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.8 kB · 81 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182{ buildPythonPackage, addDriverRunpath, 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..." addDriverRunpath "$lib" done '';
# Requires access to libcuda.so.1 which is provided by the driver doCheck = false;
checkPhase = '' py.test '';
nativeBuildInputs = [ addDriverRunpath ];
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 ]; };}