Unverified Commit 958ebc0e authored by Yuting Jiang's avatar Yuting Jiang Committed by GitHub
Browse files

Benchmarks: Fix Bug - fix bug of microbenmark building cublas and cudnn for...

Benchmarks: Fix Bug - fix bug of microbenmark building cublas and cudnn for amd in build pipeline (#166)

**Description**
Fix bug of microbenmark building cublas and cudnn for amd

**Major Revision**
- remove cuda LANGUAGES in project()
- check CUDAToolkit quiet and then build if found
parent 34cd2e8c
...@@ -2,30 +2,31 @@ ...@@ -2,30 +2,31 @@
# Licensed under the MIT License. # Licensed under the MIT License.
cmake_minimum_required(VERSION 3.18) cmake_minimum_required(VERSION 3.18)
project(cublas_benchmark LANGUAGES CUDA CXX) project(cublas_benchmark LANGUAGES CXX)
include(../cuda_common.cmake) include(../cuda_common.cmake)
find_package(CUDAToolkit QUIET)
if(CUDAToolkit_FOUND)
set(SRC "cublas_helper.cpp" CACHE STRING "source file")
set(TARGET_NAME "cublas_function" CACHE STRING "target name")
SET(SRC "cublas_helper.cpp" CACHE STRING "source file") set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${NVCC_ARCHS_SUPPORTED}")
SET(TARGET_NAME "cublas_function" CACHE STRING "target name") add_library(${TARGET_NAME} SHARED ${SRC})
link_directories( ${CUDAToolkit_LIBRARY_DIR} ${CUDAToolkit_TARGET_DIR})
include_directories( ${CUDAToolkit_INCLUDE_DIRS})
find_package(CUDAToolkit REQUIRED) include(FetchContent)
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${NVCC_ARCHS_SUPPORTED}") FetchContent_Declare(json
add_library(${TARGET_NAME} SHARED ${SRC}) GIT_REPOSITORY https://github.com/ArthurSonzogni/nlohmann_json_cmake_fetchcontent
link_directories( ${CUDAToolkit_LIBRARY_DIR} ${CUDAToolkit_TARGET_DIR}) GIT_TAG v3.7.3)
include_directories( ${CUDAToolkit_INCLUDE_DIRS}) FetchContent_GetProperties(json)
if(NOT json_POPULATED)
FetchContent_Populate(json)
add_subdirectory(${json_SOURCE_DIR} ${json_BINARY_DIR} EXCLUDE_FROM_ALL)
endif()
include(FetchContent) add_executable(cublas_benchmark cublas_test.cpp)
FetchContent_Declare(json target_link_libraries(cublas_benchmark ${TARGET_NAME} nlohmann_json::nlohmann_json CUDA::cudart CUDA::cublas)
GIT_REPOSITORY https://github.com/ArthurSonzogni/nlohmann_json_cmake_fetchcontent
GIT_TAG v3.7.3)
FetchContent_GetProperties(json)
if(NOT json_POPULATED)
FetchContent_Populate(json)
add_subdirectory(${json_SOURCE_DIR} ${json_BINARY_DIR} EXCLUDE_FROM_ALL)
endif()
add_executable(cublas_benchmark cublas_test.cpp)
target_link_libraries(cublas_benchmark ${TARGET_NAME} nlohmann_json::nlohmann_json CUDA::cudart CUDA::cublas)
install(TARGETS cublas_benchmark ${TARGET_NAME} RUNTIME DESTINATION bin LIBRARY DESTINATION lib) install(TARGETS cublas_benchmark ${TARGET_NAME} RUNTIME DESTINATION bin LIBRARY DESTINATION lib)
endif()
...@@ -2,32 +2,33 @@ ...@@ -2,32 +2,33 @@
# Licensed under the MIT License. # Licensed under the MIT License.
cmake_minimum_required(VERSION 3.18) cmake_minimum_required(VERSION 3.18)
project(cudnn_benchmark LANGUAGES CUDA CXX) project(cudnn_benchmark LANGUAGES CXX)
include(../cuda_common.cmake) include(../cuda_common.cmake)
find_package(CUDAToolkit QUIET)
if(CUDAToolkit_FOUND)
set(SRC "cudnn_helper.cpp" CACHE STRING "source file")
set(TARGET_NAME "cudnn_function" CACHE STRING "target name")
SET(SRC "cudnn_helper.cpp" CACHE STRING "source file") set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${NVCC_ARCHS_SUPPORTED}")
SET(TARGET_NAME "cudnn_function" CACHE STRING "target name") add_library(${TARGET_NAME} SHARED ${SRC})
link_directories( ${CUDAToolkit_LIBRARY_DIR} ${CUDAToolkit_TARGET_DIR})
include_directories( ${CUDAToolkit_INCLUDE_DIRS})
find_library(CUDNN_LIBRARY cudnn
HINTS ${CUDAToolkit_ROOT_DIR}
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
find_package(CUDAToolkit REQUIRED) include(FetchContent)
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${NVCC_ARCHS_SUPPORTED}") FetchContent_Declare(json
add_library(${TARGET_NAME} SHARED ${SRC}) GIT_REPOSITORY https://github.com/ArthurSonzogni/nlohmann_json_cmake_fetchcontent
link_directories( ${CUDAToolkit_LIBRARY_DIR} ${CUDAToolkit_TARGET_DIR}) GIT_TAG v3.7.3)
include_directories( ${CUDAToolkit_INCLUDE_DIRS}) FetchContent_GetProperties(json)
find_library(CUDNN_LIBRARY cudnn if(NOT json_POPULATED)
HINTS ${CUDAToolkit_ROOT_DIR} FetchContent_Populate(json)
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64) add_subdirectory(${json_SOURCE_DIR} ${json_BINARY_DIR} EXCLUDE_FROM_ALL)
endif()
include(FetchContent) add_executable(cudnn_benchmark cudnn_test.cpp)
FetchContent_Declare(json target_link_libraries(cudnn_benchmark ${TARGET_NAME} nlohmann_json::nlohmann_json CUDA::cudart ${CUDNN_LIBRARY})
GIT_REPOSITORY https://github.com/ArthurSonzogni/nlohmann_json_cmake_fetchcontent install(TARGETS cudnn_benchmark ${TARGET_NAME} RUNTIME DESTINATION bin LIBRARY DESTINATION lib)
GIT_TAG v3.7.3)
FetchContent_GetProperties(json)
if(NOT json_POPULATED)
FetchContent_Populate(json)
add_subdirectory(${json_SOURCE_DIR} ${json_BINARY_DIR} EXCLUDE_FROM_ALL)
endif() endif()
add_executable(cudnn_benchmark cudnn_test.cpp)
target_link_libraries(cudnn_benchmark ${TARGET_NAME} nlohmann_json::nlohmann_json CUDA::cudart ${CUDNN_LIBRARY})
install(TARGETS cudnn_benchmark ${TARGET_NAME} RUNTIME DESTINATION bin LIBRARY DESTINATION lib)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment