cupy: 13.0->13.2, patch to find libcudart-static

(cherry picked from commit 6d785b5787)
This commit is contained in:
cfhammill
2024-07-03 14:33:52 -04:00
committed by FliegendeWurst
parent 04b7d6ee7b
commit 1eafca5d79

View File

@@ -1,8 +1,8 @@
{
lib,
buildPythonPackage,
fetchPypi,
cython,
fetchFromGitHub,
cython_0,
fastrlock,
numpy,
wheel,
@@ -13,6 +13,7 @@
addOpenGLRunpath,
pythonOlder,
symlinkJoin,
fetchpatch
}:
let
@@ -46,11 +47,22 @@ buildPythonPackage rec {
disabled = pythonOlder "3.7";
src = fetchPypi {
inherit pname version;
hash = "sha256-5NvSsu1BWaXMDA+YpxCgFJUOssFu60VelWEo87O9DVE=";
src = fetchFromGitHub {
owner = "cupy";
repo = "cupy";
rev = "v13.2.0";
hash = "sha256-vZAtpIZztmsYeJeuq7yl7kgZse2azrIM3efHDmUswJI=";
fetchSubmodules = true;
};
patches = [
(fetchpatch {
url =
"https://github.com/cfhammill/cupy/commit/67526c756e4a0a70f0420bf0e7f081b8a35a8ee5.patch";
hash = "sha256-WZgexBdM9J0ep5s+9CGZriVq0ZidCRccox+g0iDDywQ=";
})
];
# See https://docs.cupy.dev/en/v10.2.0/reference/environment.html. Seting both
# CUPY_NUM_BUILD_JOBS and CUPY_NUM_NVCC_THREADS to NIX_BUILD_CORES results in
# a small amount of thrashing but it turns out there are a large number of
@@ -65,7 +77,7 @@ buildPythonPackage rec {
setuptools
wheel
addOpenGLRunpath
cython
cython_0
cudaPackages.cuda_nvcc
];
@@ -78,7 +90,6 @@ buildPythonPackage rec {
NVCC = "${lib.getExe cudaPackages.cuda_nvcc}"; # FIXME: splicing/buildPackages
CUDA_PATH = "${cudatoolkit-joined}";
LDFLAGS = "-L${cudaPackages.cuda_cudart}/lib/stubs";
propagatedBuildInputs = [
fastrlock