mfem: add missing cublas for cuda support (#35608)

This commit is contained in:
Aaron Black 2023-03-08 06:14:46 -08:00 committed by GitHub
parent 0d4f9b26b8
commit 81765e0278
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -651,7 +651,7 @@ def find_optional_library(name, prefix):
all_hypre_libs = hypre.libs + hypre["lapack"].libs + hypre["blas"].libs all_hypre_libs = hypre.libs + hypre["lapack"].libs + hypre["blas"].libs
hypre_gpu_libs = "" hypre_gpu_libs = ""
if "+cuda" in hypre: if "+cuda" in hypre:
hypre_gpu_libs = " -lcusparse -lcurand" hypre_gpu_libs = " -lcusparse -lcurand -lcublas"
elif "+rocm" in hypre: elif "+rocm" in hypre:
hypre_rocm_libs = LibraryList([]) hypre_rocm_libs = LibraryList([])
if "^rocsparse" in hypre: if "^rocsparse" in hypre: