diff --git a/nix-support/cutensor.nix b/nix-support/cutensor.nix index 9e44e9f8..cd188e7a 100644 --- a/nix-support/cutensor.nix +++ b/nix-support/cutensor.nix @@ -2,17 +2,17 @@ let package_map = { x86_64-linux = fetchzip { - url = "https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/linux-x86_64/libcutensor-linux-x86_64-2.0.1.2-archive.tar.xz"; - sha256 = sha256:OPn5577DBwzcGdSovv5tMv+x3wW2QgF5XVmRgm5y148=; + url = "https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/linux-x86_64/libcutensor-linux-x86_64-2.0.2.5-archive.tar.xz"; + sha256 = sha256:06ygtOh3potihT0bm0lhytKpf7zWOA8Ap9rATNvQdTk=; }; aarch64-linux = fetchzip { - url = "https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/linux-sbsa/libcutensor-linux-sbsa-2.0.1.2-archive.tar.xz"; - sha256 = sha256:vTEQ7uJYrEC1Q55DgCO5/PXTm0f34ndqezotnqq6E00=; + url = "https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/linux-sbsa/libcutensor-linux-sbsa-2.0.2.5-archive.tar.xz"; + sha256 = sha256:CRewkAlXYklDnj6iF6eGv0C8yxHVDOdj+mXqm2dN+po=; }; }; in pkgs.stdenv.mkDerivation { pname = "libcutensor"; - version = "2.0.1.2"; + version = "2.0.2.5"; outputs = [ "out" "lib" "dev" "static" ]; src = package_map.${pkgs.system}; installPhase = let diff --git a/nix-support/pycuquantum.patch b/nix-support/pycuquantum.patch index fe1a782a..1f4f0b07 100644 --- a/nix-support/pycuquantum.patch +++ b/nix-support/pycuquantum.patch @@ -41,7 +41,7 @@ index cd54946..e46633e 100644 cdef void* load_library() except* nogil: cdef void* handle - handle = dlopen("libcutensornet.so.2", RTLD_NOW | RTLD_GLOBAL) -+ handle = dlopen("@cuda_bundle/lib/libcutensornet.so.2", RTLD_NOW | RTLD_GLOBAL) ++ handle = dlopen("@cuda_bundle@/lib/libcutensornet.so.2", RTLD_NOW | RTLD_GLOBAL) if handle == NULL: with gil: err_msg = dlerror()