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
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +4 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -34,6 +34,10 @@ add_library(
    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)