Lines Matching refs:CUDA

28 # Find CUDA.
29 find_package(CUDA)
32     "Caffe2: CUDA cannot be found. Depending on whether you are building "
39 # Enable CUDA language support
42 # Must be done before CUDA language is enabled, see
47 enable_language(CUDA)
64   message(FATAL_ERROR "Found two conflicting CUDA versions:\n"
69 message(STATUS "Caffe2: CUDA detected: " ${CUDA_VERSION})
70 message(STATUS "Caffe2: CUDA nvcc is: " ${CUDA_NVCC_EXECUTABLE})
71 message(STATUS "Caffe2: CUDA toolkit directory: " ${CUDA_TOOLKIT_ROOT_DIR})
73   message(FATAL_ERROR "PyTorch requires CUDA 11.0 or above.")
77   # Sometimes, we may mismatch nvcc with the CUDA headers we are
110       # Force CUDA to be processed for again next time
118 …message(FATAL_ERROR "FindCUDA says CUDA version is ${cuda_version_from_findcuda} (usually determin…
119         "but the CUDA headers say the version is ${cuda_version_from_header}.  This often occurs "
128 # ---[ CUDA libraries wrapper
150 # is set. This path is where sane CUDA installations have their static
158     CUDA::cuda_driver)
165         CUDA::cudart_static)
169         CUDA::cudart)
182   set_property(TARGET torch::nvtoolsext PROPERTY INTERFACE_LINK_LIBRARIES CUDA::nvToolsExt)
192         CUDA::cublas CUDA::cublasLt)
195         CUDA::cudart_static rt)
199         CUDA::cublas CUDA::cublasLt)
273           CUDA::cuFile_static)
277           CUDA::cuFile)
288         CUDA::curand_static)
292         CUDA::curand)
300         CUDA::cufft_static_nocallback)
304         CUDA::cufft)
311     CUDA::nvrtc caffe2::cuda)
331 message(STATUS "Added CUDA NVCC flags for: ${NVCC_FLAGS_EXTRA}")