if(USE_PROF)
  set(Caffe2_CONTRIB_PROF_CPU_SRCS
      "${CMAKE_CURRENT_SOURCE_DIR}/prof_dag_stats_op.cc"
  )
  set(Caffe2_CONTRIB_PROF_GPU_SRCS
      "${CMAKE_CURRENT_SOURCE_DIR}/cuda_profile_ops.cc"
  )

  file(GLOB_RECURSE Caffe2_CONTRIB_PROF_CPU_CPP_TEST_SRCS *test.cc)

  set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} ${Caffe2_CONTRIB_PROF_CPU_SRCS} PARENT_SCOPE)
  set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS}
      ${Caffe2_CONTRIB_PROF_CPU_CPP_TEST_SRCS} PARENT_SCOPE)
  set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${Caffe2_CONTRIB_PROF_GPU_SRCS} PARENT_SCOPE)

endif()