include(ExternalProject)

set(CUB_INCLUDE_DIR ${THIRD_PARTY_DIR}/cub/include)
set(CUB_BUILD_INCLUDE ${CMAKE_CURRENT_BINARY_DIR}/cub/src/cub/cub)

set(CUB_URL https://github.com/NVIDIA/cub/archive/refs/tags/1.11.0.tar.gz)
use_mirror(VARIABLE CUB_URL URL ${CUB_URL})

if(THIRD_PARTY)

  ExternalProject_Add(
    cub
    PREFIX cub
    URL ${CUB_URL}
    URL_MD5 97196a885598e40592100e1caaf3d5ea
    CONFIGURE_COMMAND ""
    BUILD_COMMAND ""
    INSTALL_COMMAND "")

  add_copy_headers_target(
    NAME
    cub
    SRC
    ${CUB_BUILD_INCLUDE}
    DST
    ${CUB_INCLUDE_DIR}/cub
    DEPS
    cub
    INDEX_FILE
    "${oneflow_cmake_dir}/third_party/header_index/cub_headers.txt")

endif(THIRD_PARTY)