try liblapack_static, may need if cuda >= 10.1

xwang233 · xwang233 · commit 3075adc8e176 · 2021-01-21T18:27:41.000-08:00
diff --git a/aten/src/ATen/CMakeLists.txt b/aten/src/ATen/CMakeLists.txt
@@ -325,6 +325,7 @@ if(USE_CUDA AND NOT USE_ROCM)
       ${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcublas_static.a
       ${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcufft_static_nocallback.a
       ${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcusolver_static.a
+      ${CUDA_TOOLKIT_ROOT_DIR}/lib64/liblapack_static.a     # needed for libcusolver_static
       )
   else()
     list(APPEND ATen_CUDA_DEPENDENCY_LIBS

Original file line number	Diff line number	Diff line change
`@@ -325,6 +325,7 @@ if(USE_CUDA AND NOT USE_ROCM)`
`325`	`325`	`${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcublas_static.a`
`326`	`326`	`${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcufft_static_nocallback.a`
`327`	`327`	`${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcusolver_static.a`
	`328`	`+ ${CUDA_TOOLKIT_ROOT_DIR}/lib64/liblapack_static.a # needed for libcusolver_static`
`328`	`329`	`)`
`329`	`330`	`else()`
`330`	`331`	`list(APPEND ATen_CUDA_DEPENDENCY_LIBS`