half.cmake 1.1 KB
Newer Older
yuguo's avatar
yuguo committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
include(ExternalProject)

set(HALF_INCLUDE_DIR ${THIRD_PARTY_DIR}/half/include)

set(HALF_URL https://github.com/Oneflow-Inc/half/archive/refs/tags/v2.1.0-fix-cuda-raise.zip)
use_mirror(VARIABLE HALF_URL URL ${HALF_URL})
set(HALF_BASE_DIR ${CMAKE_CURRENT_BINARY_DIR}/half/src/half)
set(HALF_URL_HASH 30b0dc289729f9e85ddf6995f2e6968f)
set(HALF_HEADERS "${HALF_BASE_DIR}/include/half.hpp")

if(THIRD_PARTY)

  ExternalProject_Add(
    half
    PREFIX half
    URL ${HALF_URL}
    URL_HASH MD5=${HALF_URL_HASH}
    UPDATE_COMMAND ""
    CONFIGURE_COMMAND ""
    BUILD_COMMAND ""
    BUILD_IN_SOURCE 1
    INSTALL_COMMAND "")

  add_custom_target(half_create_header_dir COMMAND ${CMAKE_COMMAND} -E make_directory
                                                   ${HALF_INCLUDE_DIR} DEPENDS half)

  add_custom_target(half_copy_headers_to_destination DEPENDS half_create_header_dir)

  foreach(header_file ${HALF_HEADERS})
    add_custom_command(
      TARGET half_copy_headers_to_destination PRE_BUILD
      COMMAND ${CMAKE_COMMAND} -E copy_if_different ${header_file} ${HALF_INCLUDE_DIR})
  endforeach()
endif(THIRD_PARTY)