Skip to content

Commit 460cfa4

Browse files
authored
Updated cutensornet to 24.03 and cutensor to 2.0.1 (#600)
1 parent 2c34e34 commit 460cfa4

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,8 +175,8 @@ endif()
175175

176176
# Get the tensor libraries if we need them
177177
if (MATX_EN_CUTENSOR)
178-
set(CUTENSORNET_VERSION 23.03.0.20)
179-
set(CUTENSOR_VERSION 1.7.0.1)
178+
set(CUTENSORNET_VERSION 24.03.0.4)
179+
set(CUTENSOR_VERSION 2.0.1.2)
180180

181181
include(cmake/FindcuTENSOR.cmake)
182182
include(cmake/FindcuTensorNet.cmake)

cmake/FindcuTensorNet.cmake

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ endif()
8181
if(NOT cuTensorNet_FOUND)
8282
message(STATUS "cuTensorNet not found. Downloading library. By continuing this download you accept to the license terms of cuQuantum SDK")
8383

84-
set(CUTENSORNET_FILENAME cuquantum-linux-x86_64-${CUTENSORNET_VERSION}-archive)
84+
set(CUTENSORNET_FILENAME cuquantum-linux-x86_64-${CUTENSORNET_VERSION}_cuda${CUDAToolkit_VERSION_MAJOR}-archive)
8585

8686
CPMAddPackage(
8787
NAME cutensornet
@@ -91,7 +91,7 @@ if(NOT cuTensorNet_FOUND)
9191
DOWNLOAD_ONLY YES
9292
)
9393

94-
set(cuTensorNet_LIBRARY ${cutensornet_SOURCE_DIR}/lib/${CUDAToolkit_VERSION_MAJOR}/libcutensornet.so)
94+
set(cuTensorNet_LIBRARY ${cutensornet_SOURCE_DIR}/lib/libcutensornet.so)
9595
set(cuTensorNet_INCLUDE_DIR ${cutensornet_SOURCE_DIR}/include)
9696

9797
set(cuTensorNet_FOUND TRUE)

docs_input/build.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ Optional Third-party Dependencies
4040
- `GoogleTest <https://github.com/google/googletest>`_ 1.11.0+ (Required to run unit tests)
4141
- `pybind11 <https://github.com/pybind/pybind11>`_ 2.6.2+ (Required for file I/O and some unit tests)
4242
- `nvbench <https://github.com/NVIDIA/nvbench>`_ Commit 1a13a2e (Required to run benchmarks)
43-
- `cutensor <https://developer.nvidia.com/cutensor>`_ 1.7.0.1+ (Required when using `einsum`)
44-
- `cutensornet <https://docs.nvidia.com/cuda/cuquantum/cutensornet>`_ 23.03.0.20+ (Required when using `einsum`)
43+
- `cutensor <https://developer.nvidia.com/cutensor>`_ 2.0.1.2+ (Required when using `einsum`)
44+
- `cutensornet <https://docs.nvidia.com/cuda/cuquantum/cutensornet>`_ 24.03.0.4+ (Required when using `einsum`)
4545

4646
Host (CPU) Support
4747
------------------

0 commit comments

Comments
 (0)