Skip to content

Commit f534340

Browse files
authored
Add NVPL libraries to pytorch installation (#130)
1 parent 44d84bd commit f534340

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

CMakeLists.txt

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,16 @@ if (CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "aarch64")
162162
set(LIBS_ARCH "aarch64")
163163
set(LIBTORCH_LIBS
164164
"libopenblas.so.0"
165+
"libnvpl_blas_core.so.0"
166+
"libnvpl_blas_ilp64_gomp.so.0"
167+
"libnvpl_blas_ilp64_seq.so.0"
168+
"libnvpl_blas_lp64_gomp.so.0"
169+
"libnvpl_blas_lp64_seq.so.0"
170+
"libnvpl_lapack_core.so.0"
171+
"libnvpl_lapack_ilp64_gomp.so.0"
172+
"libnvpl_lapack_ilp64_seq.so.0"
173+
"libnvpl_lapack_lp64_gomp.so.0"
174+
"libnvpl_lapack_lp64_seq.so.0"
165175
)
166176
else()
167177
set(LIBS_ARCH "x86_64")

0 commit comments

Comments
 (0)