Add XLA compiler frontend.
* Names it iree-import-xla (backwards from iree-tf-import) because Ben asked me to swap the order. I'll followup with a rename of the TF side.
* Does not test HLO pbtext because I couldn't find a reliable way to generate it from OSS tooling.
* Fixes a little bug in the bazel integration that was keeping more than one target from building.
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 9add1e0..0aaf142 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -45,15 +45,24 @@
option(IREE_BUILD_DEBUGGER "Builds the IREE debugger app." OFF)
option(IREE_BUILD_JAVA_BINDINGS "Builds the IREE java bindings." OFF)
option(IREE_BUILD_EXPERIMENTAL "Builds experimental projects." OFF)
-option(IREE_BUILD_TENSORFLOW_COMPILER "Builds TensorFlow compiler." OFF)
+option(IREE_BUILD_TENSORFLOW_COMPILER "Builds TensorFlow compiler frontend." OFF)
+option(IREE_BUILD_XLA_COMPILER "Builds TensorFlow XLA compiler frontend." OFF)
set(IREE_HAL_DRIVERS_TO_BUILD "all"
CACHE STRING "Semicolon-separated list of HAL drivers to build, or \"all\".")
set(IREE_TARGET_BACKENDS_TO_BUILD "all"
CACHE STRING "Semicolon-separated list of target backends to build, or \"all\".")
+# Master enable for tensorflow build support.
+# Note that this is a normal CMake variable used to gate build features (not
+# a cache variable that is user-settable).
+set(IREE_ENABLE_TENSORFLOW OFF)
+if(${IREE_BUILD_TENSORFLOW_COMPILER} OR ${IREE_BUILD_XLA_COMPILER})
+ set(IREE_ENABLE_TENSORFLOW ON)
+endif()
+
# Default python bindings to enabled for some features.
-if(${IREE_BUILD_TENSORFLOW_COMPILER})
+if(${IREE_ENABLE_TENSORFLOW})
option(IREE_BUILD_PYTHON_BINDINGS "Builds the IREE python bindings" ON)
else()
option(IREE_BUILD_PYTHON_BINDINGS "Builds the IREE python bindings" OFF)
@@ -282,8 +291,7 @@
#-------------------------------------------------------------------------------
if(${IREE_BUILD_COMPILER} OR
- ${IREE_BUILD_PYTHON_BINDINGS} OR
- ${IREE_BUILD_TENSORFLOW_COMPILER})
+ ${IREE_BUILD_PYTHON_BINDINGS})
find_package(Python3 COMPONENTS Interpreter REQUIRED)
endif()
@@ -404,7 +412,7 @@
# Depends on python configuration.
#-------------------------------------------------------------------------------
-if(${IREE_BUILD_TENSORFLOW_COMPILER})
+if(${IREE_ENABLE_TENSORFLOW})
include(configure_bazel)
iree_configure_bazel()
endif()
@@ -531,7 +539,7 @@
add_subdirectory(experimental)
endif()
-if(${IREE_BUILD_TENSORFLOW_COMPILER})
+if(${IREE_ENABLE_TENSORFLOW})
add_subdirectory(integrations/tensorflow)
endif()
