caffe2/observers/CMakeLists.txt

if(USE_OBSERVERS)
  message(STATUS "Include Observer library")
  set(GLOB profile_observer_files profile_observer_*.cc)
  set(Caffe2_CONTRIB_OBSERVERS_CPU_SRC
    "${CMAKE_CURRENT_SOURCE_DIR}/time_observer.cc"
    "${CMAKE_CURRENT_SOURCE_DIR}/runcnt_observer.cc"
  )
  set(Caffe2_CONTRIB_OBSERVERS_GPU_SRC
    "${CMAKE_CURRENT_SOURCE_DIR}/profile_observer_gpu.cc"
  )

  set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} ${Caffe2_CONTRIB_OBSERVERS_CPU_SRC})
  set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)

  set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${Caffe2_CONTRIB_OBSERVERS_GPU_SRC})
  set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} PARENT_SCOPE)

  # ---[ CPU test files
  file(GLOB tmp *_test.cc)
  set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} ${tmp})
  set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE)
  exclude(Caffe2_CPU_TEST_SRCS "${Caffe2_CPU_TEST_SRCS}" ${profile_observer_files})

  # ---[ GPU test files
  set(Caffe2_GPU_TEST_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/profile_observer_test.cc")
endif()