| 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() |
