From f3ec1ddaee795871715c029f97badcb1a37e3bb0 Mon Sep 17 00:00:00 2001 From: Emmanuel Schmidbauer Date: Mon, 28 Aug 2023 13:11:23 -0400 Subject: [PATCH] fix build: default to lib instead of lib64 --- cmake/cuda.cmake | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index f4111d8ea6..a42124419d 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -12,9 +12,9 @@ if(CUDA_FOUND) # set cuda runtime and driver lib # override cublas and curand because the FindCUDA module may not find the correct libs - set(CUDADRV_LIBRARIES ${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs/libcuda${LIBEXT}) - set(CUDA_CUBLAS_LIBRARIES ${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcublas${LIBEXT}) - set(CUDA_curand_LIBRARY ${CUDA_TOOLKIT_ROOT_DIR}/lib64/libcurand${LIBEXT}) + set(CUDADRV_LIBRARIES ${CUDA_TOOLKIT_ROOT_DIR}/lib/stubs/libcuda${LIBEXT}) + set(CUDA_CUBLAS_LIBRARIES ${CUDA_TOOLKIT_ROOT_DIR}/lib/libcublas${LIBEXT}) + set(CUDA_curand_LIBRARY ${CUDA_TOOLKIT_ROOT_DIR}/lib/libcurand${LIBEXT}) list(APPEND FLEXFLOW_EXT_LIBRARIES ${CUDADRV_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES}