Merge pull request cupy#8282 from kmaehashi/cutensor-2.0.1

Bump to cuTENSOR 2.0.1
chainer-ci · Apr 15, 2024 · b484198 · b484198
1 parent 5e9795b
commit b484198
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cupyx/tools/install_library.py b/cupyx/tools/install_library.py
@@ -107,9 +107,9 @@ def __make_cutensor_record(
 
 def _make_cutensor_record(cuda_version):
     return __make_cutensor_record(
-        cuda_version, '2.0.0',
-        'libcutensor-linux-x86_64-2.0.0.7-archive.tar.xz',
-        'libcutensor-windows-x86_64-2.0.0.7-archive.zip')
+        cuda_version, '2.0.1',
+        'libcutensor-linux-x86_64-2.0.1.2-archive.tar.xz',
+        'libcutensor-windows-x86_64-2.0.1.2-archive.zip')
 
 
 _cutensor_records.append(_make_cutensor_record('12.x'))