mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
c373387709
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62445 PyTorch currently uses the old style of compiling CUDA in CMake which is just a bunch of scripts in `FindCUDA.cmake`. Newer versions support CUDA natively as a language just like C++ or C. Test Plan: Imported from OSS Reviewed By: ejguan Differential Revision: D31503350 fbshipit-source-id: 2ee817edc9698531ae1b87eda3ad271ee459fd55
20 lines
494 B
CMake
20 lines
494 B
CMake
# Try to find the CUB library and headers.
|
|
# CUB_FOUND - system has CUB
|
|
# CUB_INCLUDE_DIRS - the CUB include directory
|
|
|
|
find_path(CUB_INCLUDE_DIR
|
|
HINTS "${CUDA_TOOLKIT_INCLUDE}"
|
|
NAMES cub/cub.cuh
|
|
DOC "The directory where CUB includes reside"
|
|
)
|
|
|
|
set(CUB_INCLUDE_DIRS ${CUB_INCLUDE_DIR})
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(CUB
|
|
FOUND_VAR CUB_FOUND
|
|
REQUIRED_VARS CUB_INCLUDE_DIR
|
|
)
|
|
|
|
mark_as_advanced(CUB_FOUND)
|