option(ENABLE_UCX"Enable building with UCX (Uniform Communication X) support"

ON)

option(NVRTC_DYNAMIC_LINKING"Link against the dynamic NVRTC libraries"OFF)

option(CUBLAS_DYNAMIC_LINKING

"Link against the dynamic cublas/cublasLt libraries"ON)

option(CURAND_DYNAMIC_LINKING"Link against the dynamic curand library"ON)

option(ENABLE_NVSHMEM"Enable building with NVSHMEM support"OFF)

option(USING_OSS_CUTLASS_LOW_LATENCY_GEMM

"Using open sourced Cutlass low latency gemm kernel"ON)

setup_cuda_architectures()

find_package(CUDAToolkit 11.2 REQUIRED COMPONENTS cudart_static cuda_driver

cublas cublasLt curand nvml)

set(CUBLAS_LIB CUDA::cublas)

set(CUBLASLT_LIB CUDA::cublasLt)

set(CURAND_LIB CUDA::curand)

set(CUDA_DRV_LIB CUDA::cuda_driver)

set(CUDA_NVML_LIB CUDA::nvml)

set(CUDA_RT_LIB CUDA::cudart_static)

set(NVPTX_LIB CUDA::nvptxcompiler_static)

set(CMAKE_CUDA_RUNTIME_LIBRARYStatic)

set(CUDA_TOOLKIT_COMPONENTS cudart_static cuda_driver nvml nvptxcompiler_static)

if(CUBLAS_DYNAMIC_LINKING)

set(CUBLAS_LIB CUDA::cublas)

set(CUBLASLT_LIB CUDA::cublasLt)

list(APPEND CUDA_TOOLKIT_COMPONENTS cublas cublasLt)

else()

if(WIN32)

message(FATAL_ERROR"Static cublas not available on windows")

endif()

message(DEBUG"Linking with static cublas libs")

set(CUBLAS_LIB CUDA::cublas_static)

set(CUBLASLT_LIB CUDA::cublasLt_static)

list(APPEND CUDA_TOOLKIT_COMPONENTS cublas_static cublasLt_static)

endif()

if(CURAND_DYNAMIC_LINKING)

set(CURAND_LIB CUDA::curand)

list(APPEND CUDA_TOOLKIT_COMPONENTS curand)

else()

if(WIN32)

message(FATAL_ERROR"Static curand not available on windows")

endif()

message(DEBUG"Linking with static curand lib")

set(CURAND_LIB CUDA::curand_static)

list(APPEND CUDA_TOOLKIT_COMPONENTS curand_static)

endif()

if(NVRTC_DYNAMIC_LINKING)

set(NVRTC_LIB CUDA::nvrtc)

set(NVRTC_BUILTINS_LIB CUDA::nvrtc_builtins)

list(APPEND CUDA_TOOLKIT_COMPONENTS nvrtc nvrtc_builtins)

else()

set(NVRTC_LIB CUDA::nvrtc_static)

set(NVRTC_BUILTINS_LIB CUDA::nvrtc_builtins_static)

list(APPEND CUDA_TOOLKIT_COMPONENTS nvrtc_static nvrtc_builtins_static)

endif()

find_package(CUDAToolkit 11.2 REQUIRED COMPONENTS${CUDA_TOOLKIT_COMPONENTS})

set(CMAKE_CUDA_RUNTIME_LIBRARYStatic)

resolve_dirs(CUDAToolkit_INCLUDE_DIRS"${CUDAToolkit_INCLUDE_DIRS}")

message(STATUS"CUDA library status:")