mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
48a35135fb
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18959 ghimport-source-id: a934163fa34cb2019732d5f49dc7290c376bf156 Differential Revision: D14831246 Pulled By: ezyang fbshipit-source-id: beb92dc4ee8c82f4c8259c081dd72e477fe7a9d0
19 lines
454 B
CMake
19 lines
454 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
|
|
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)
|