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)

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)

if(CUBLAS_DYNAMIC_LINKING)

set(CUBLAS_LIB CUDA::cublas)

set(CUBLASLT_LIB CUDA::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)

endif()

if(CURAND_DYNAMIC_LINKING)

set(CURAND_LIB CUDA::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)

endif()

set(CUDA_DRV_LIB CUDA::cuda_driver)

set(CUDA_NVML_LIB CUDA::nvml)

set(CUDA_RT_LIB CUDA::cudart_static)