File tree Expand file tree Collapse file tree 3 files changed +6
-6
lines changed
Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -175,8 +175,8 @@ endif()
175175
176176# Get the tensor libraries if we need them
177177if (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 )
Original file line number Diff line number Diff line change @@ -81,7 +81,7 @@ endif()
8181if (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 )
Original file line number Diff line number Diff 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
4646Host (CPU) Support
4747------------------
You can’t perform that action at this time.
0 commit comments