[MLGO] Correct protobuf path

diff --git a/llvm/CMakeLists.txt b/llvm/CMakeLists.txt
index 8f4d897..37218cd 100644 (file)
--- a/llvm/CMakeLists.txt
+++ b/llvm/CMakeLists.txt
@@ -803,7 +803,7 @@ if (TENSORFLOW_C_LIB_PATH)
     endif()
     string(REGEX MATCH "Location: ([^\n]*\n)" TF_PIP_LOC "${TF_PIP_OUT}")
     string(REPLACE "Location: " "" TF_PIP ${TF_PIP_LOC})
-    set(TF_PROTO_HEADERS ${TF_PIP}/include)
+    set(TF_PROTO_HEADERS ${TF_PIP}/tensorflow/include)
   endif()
   include_directories(${TF_PROTO_HEADERS})
   add_definitions("-DGOOGLE_PROTOBUF_NO_RTTI")