Skip to content

Commit 207c4d4

Browse files
committed
[MLGO] Strip TF_PIP cmake variable
This should fix build breaks for 'development' mode. The other modes were unaffected - 'release' because it doesn't use TFUtils.cpp, and the mixed mode because the AOT compiled code brings in the necessary include dirs anyway.
1 parent ea45235 commit 207c4d4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

llvm/CMakeLists.txt

+3-1
Original file line numberDiff line numberDiff line change
@@ -803,8 +803,10 @@ if (TENSORFLOW_C_LIB_PATH)
803803
endif()
804804
string(REGEX MATCH "Location: ([^\n]*\n)" TF_PIP_LOC "${TF_PIP_OUT}")
805805
string(REPLACE "Location: " "" TF_PIP ${TF_PIP_LOC})
806-
set(TF_PROTO_HEADERS ${TF_PIP}/tensorflow/include)
806+
string(STRIP ${TF_PIP} TF_PIP)
807+
set(TF_PROTO_HEADERS "${TF_PIP}/tensorflow/include")
807808
endif()
809+
message(STATUS "Using Tensorflow headers under: ${TF_PROTO_HEADERS}")
808810
include_directories(${TF_PROTO_HEADERS})
809811
add_definitions("-DGOOGLE_PROTOBUF_NO_RTTI")
810812
add_definitions("-D_GLIBCXX_USE_CXX11_ABI=0")

0 commit comments

Comments
 (0)