caffe2/utils/CMakeLists.txt

list(APPEND Caffe2_CPU_SRCS
  utils/proto_convert.cc
  utils/proto_wrap.cc
  utils/proto_utils.cc
  utils/murmur_hash3.cc
  utils/smart_tensor_printer.cc
  utils/signal_handler.cc
  utils/string_utils.cc
  utils/threadpool/ThreadPool.cc
  utils/cpuid.cc
  utils/bench_utils.cc
  utils/math_cpu.cc
  utils/math_utils.cc
  utils/thread_name.cc)

# ---[ threadpool/pthreadpool* is a local modification of the NNPACK
# pthreadpool with a very similar interface. Neither NNPACK, nor this
# thread pool supports Windows.
if (NOT MSVC)
  set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS}
          utils/threadpool/pthreadpool.cc
          utils/threadpool/pthreadpool_impl.cc
          )
endif()

set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS}
        utils/math_gpu.cu
        )

set(Caffe2_HIP_SRCS ${Caffe2_HIP_SRCS}
        utils/hip/math_hip.cc
        )

set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS}
        utils/fixed_divisor_test.cc
        utils/math_test.cc
        utils/fatal_signal_asan_no_sig_test.cc
        utils/simple_queue_test.cc
        utils/proto_utils_test.cc
        utils/cpuid_test.cc
        utils/smart_tensor_printer_test.cc
        utils/cast_test.cc
        )

set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS}
        utils/math_gpu_test.cc
        )

set(Caffe2_HIP_TEST_SRCS ${Caffe2_HIP_TEST_SRCS}
        utils/hip/math_hip_test.cc
        utils/hip/math_blas_hip_test.cc
        )

# TODO Remove the CMake_xxx variables above and add them to the variables for the local library target below instead

set(LIB_SOURCES_CPU
        dummy.cpp
        )

set(LIB_SOURCES_GPU
        dummy.cpp
        )

set(LIB_SOURCES_HIP
        dummy.cpp
        )

set(TEST_SOURCES_CPU
        )

set(LIB_SOURCES_GPU
        dummy.cpp
        )

set(TEST_SOURCES_HIP
        dummy.cpp
        )

add_library(c10_utils_cpu OBJECT ${LIB_SOURCES_CPU})
target_enable_style_warnings("c10_utils_cpu")

add_library(c10_utils_gpu OBJECT ${LIB_SOURCES_GPU})
target_enable_style_warnings("c10_utils_gpu")

add_library(c10_utils_hip OBJECT ${LIB_SOURCES_HIP})
target_enable_style_warnings("c10_utils_hip")

if(BUILD_TEST)
  add_executable(c10_utils_cpu_test ${TEST_SOURCES_CPU} $<TARGET_OBJECTS:c10_utils_cpu>)
  add_test(NAME c10_utils_cpu_test COMMAND $<TARGET_FILE:c10_utils_cpu_test>)
  target_enable_style_warnings(c10_utils_cpu_test)
  target_link_libraries(c10_utils_cpu_test gtest_main)
  if(INSTALL_TEST)
    install(TARGETS c10_utils_cpu_test DESTINATION test)
  endif()

  add_executable(c10_utils_gpu_test ${TEST_SOURCES_GPU} $<TARGET_OBJECTS:c10_utils_gpu>)
  add_test(NAME c10_utils_gpu_test COMMAND $<TARGET_FILE:c10_utils_gpu_test>)
  target_enable_style_warnings(c10_utils_gpu_test)
  target_link_libraries(c10_utils_gpu_test gtest_main)
  if(INSTALL_TEST)
    install(TARGETS c10_utils_gpu_test DESTINATION test)
  endif()

  add_executable(c10_utils_hip_test ${TEST_SOURCES_HIP} $<TARGET_OBJECTS:c10_utils_hip>)
  add_test(NAME c10_utils_hip_test COMMAND $<TARGET_FILE:c10_utils_hip_test>)
  target_enable_style_warnings(c10_utils_hip_test)
  target_link_libraries(c10_utils_hip_test gtest_main)
  if(INSTALL_TEST)
    install(TARGETS c10_utils_hip_test DESTINATION test)
  endif()
endif()


# TODO Once all source files are defined inside the local c10_utils_xxx targets,
# it should be the job of the parent CMakeLists.txt to decide what to do with the target (i.e. link it to caffe2)
# instead of us locally adding it to Caffe2_xxx variables.
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} PARENT_SCOPE)
set(Caffe2_HIP_SRCS ${Caffe2_HIP_SRCS} PARENT_SCOPE)
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE)
set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS} PARENT_SCOPE)
set(Caffe2_HIP_TEST_SRCS ${Caffe2_HIP_TEST_SRCS} PARENT_SCOPE)