Add NVPL libraries to pytorch installation (#130) (#131)

triton-inference-server · May 29, 2024 · 2704676 · 2704676
1 parent 44d84bd
commit 2704676
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -162,6 +162,16 @@ if (CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "aarch64")
   set(LIBS_ARCH "aarch64")
   set(LIBTORCH_LIBS
       "libopenblas.so.0"
+      "libnvpl_blas_core.so.0"
+      "libnvpl_blas_ilp64_gomp.so.0"
+      "libnvpl_blas_ilp64_seq.so.0"
+      "libnvpl_blas_lp64_gomp.so.0"
+      "libnvpl_blas_lp64_seq.so.0"
+      "libnvpl_lapack_core.so.0"
+      "libnvpl_lapack_ilp64_gomp.so.0"
+      "libnvpl_lapack_ilp64_seq.so.0"
+      "libnvpl_lapack_lp64_gomp.so.0"
+      "libnvpl_lapack_lp64_seq.so.0"
   )
 else()
   set(LIBS_ARCH "x86_64")