Fixed pycuquantum patch, updated cutensor for cutensornet compat

CQCL · Oct 23, 2024 · ac23e12 · ac23e12
1 parent 0cc09b2
commit ac23e12
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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()