| file(GLOB Detectron_CPU_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/*.cc) | |
| file(GLOB Detectron_GPU_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/*.cu) | |
| # Note(ilijar): Since Detectron ops currently have no | |
| # CPU implementation, we only build GPU ops for now. | |
| if (USE_CUDA) | |
| CUDA_ADD_LIBRARY( | |
| caffe2_detectron_ops_gpu SHARED | |
| ${Detectron_CPU_SRCS} | |
| ${Detectron_GPU_SRCS}) | |
| target_link_libraries(caffe2_detectron_ops_gpu caffe2_gpu) | |
| install(TARGETS caffe2_detectron_ops_gpu DESTINATION lib) | |
| endif() |