Specify caffe2 include directories in custom ops cmake

Reviewed By: rbgirshick Differential Revision: D10857403 fbshipit-source-id: 5ea53ece3418e8f6dcb20df3d3521d661a7380a3

Specify caffe2 include directories in custom ops cmake
Reviewed By: rbgirshick Differential Revision: D10857403 fbshipit-source-id: 5ea53ece3418e8f6dcb20df3d3521d661a7380a3
b2474ff2 · Ilija Radosavovic · Facebook Github Bot · c003e5af · b2474ff2
Commit b2474ff2 authored Nov 03, 2018 by Ilija Radosavovic Committed by Facebook Github Bot Nov 03, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

CMakeLists.txt CMakeLists.txt +6 -2

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -31,8 +31,12 @@ file(GLOB CUSTOM_OPS_GPU_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/detectron/ops/*.cu)

 # Install custom CPU ops lib.
 add_library(
-     caffe2_detectron_custom_ops SHARED
-     ${CUSTOM_OPS_CPU_SRCS})
+    caffe2_detectron_custom_ops SHARED
+    ${CUSTOM_OPS_CPU_SRCS})
+
+target_include_directories(
+    caffe2_detectron_custom_ops PRIVATE
+    ${CAFFE2_INCLUDE_DIRS})

 target_link_libraries(caffe2_detectron_custom_ops caffe2_library)
 install(TARGETS caffe2_detectron_custom_ops DESTINATION lib)