Make TF tests optional in python CI build script. (#2249)

This should be optional for users relying on these CI scripts to build
StableHLO python bindings in other projects / locally.

.github/workflows/buildAndTestCMake.yml

@@ -98,3 +98,4 @@ jobs:
       env:
           CMAKE_BUILD_TYPE: Release
           STABLEHLO_ENABLE_BINDINGS_PYTHON: ON
+          STABLEHLO_ENABLE_PYTHON_TF_TESTS: ON

.github/workflows/publishWheelRelease.yml

@@ -88,6 +88,7 @@ jobs:
       env:
           CMAKE_BUILD_TYPE: Release
           STABLEHLO_ENABLE_BINDINGS_PYTHON: ON
+          STABLEHLO_ENABLE_PYTHON_TF_TESTS: ON
 
     - name: Build and Test Python Wheel
       shell: bash

CMakeLists.txt

@@ -44,6 +44,7 @@ endif()
 #-------------------------------------------------------------------------------
 option(STABLEHLO_BUILD_EMBEDDED "Build StableHLO as part of another project" OFF)
 option(STABLEHLO_ENABLE_BINDINGS_PYTHON "Enables StableHLO Python bindings" OFF)
+option(STABLEHLO_ENABLE_PYTHON_TF_TESTS "Enables StableHLO to SavedModel tests requiring TF" OFF)
 option(STABLEHLO_ENABLE_STRICT_BUILD "Build StableHLO with strict warnings and warnings as errors" OFF)
 option(STABLEHLO_ENABLE_SANITIZER "Enable a sanitizer [OFF, address]" OFF)
 option(STABLEHLO_ENABLE_SPLIT_DWARF "Enable split DWARF if the platform supports it" OFF)

build_tools/github_actions/ci_build_cmake.sh

@@ -35,6 +35,8 @@ CMAKE_BUILD_TYPE="${CMAKE_BUILD_TYPE:-RelWithDebInfo}"
 
 # Turn on building Python bindings
 STABLEHLO_ENABLE_BINDINGS_PYTHON="${STABLEHLO_ENABLE_BINDINGS_PYTHON:-OFF}"
+# Turn on running SavedModel Python tests requiring TF dependency
+STABLEHLO_ENABLE_PYTHON_TF_TESTS="${STABLEHLO_ENABLE_PYTHON_TF_TESTS:-OFF}"
 # Turn on building Sanitizers
 # Note: This is not congruent with building python bindings
 STABLEHLO_ENABLE_SANITIZER="${STABLEHLO_ENABLE_SANITIZER:-OFF}"
@@ -56,7 +58,8 @@ cmake -GNinja \
   -DSTABLEHLO_ENABLE_STRICT_BUILD=ON \
   -DCMAKE_PLATFORM_NO_VERSIONED_SONAME:BOOL=ON \
   -DSTABLEHLO_ENABLE_SANITIZER="$STABLEHLO_ENABLE_SANITIZER" \
-  -DSTABLEHLO_ENABLE_BINDINGS_PYTHON="$STABLEHLO_ENABLE_BINDINGS_PYTHON"
+  -DSTABLEHLO_ENABLE_BINDINGS_PYTHON="$STABLEHLO_ENABLE_BINDINGS_PYTHON" \
+  -DSTABLEHLO_ENABLE_PYTHON_TF_TESTS="$STABLEHLO_ENABLE_PYTHON_TF_TESTS"
 
 # Build and Test StableHLO
 cd "$STABLEHLO_BUILD_DIR" || exit

stablehlo/integrations/python/tests/CMakeLists.txt

@@ -27,7 +27,11 @@ endfunction()
 add_stablehlo_python_test(stablehlo-python-chlo chlo.py)
 add_stablehlo_python_test(stablehlo-python-smoketest smoketest.py)
 add_stablehlo_python_test(stablehlo-python-stablehlo stablehlo.py)
-add_stablehlo_python_test(stablehlo-python-stablehlo-to-saved-model stablehlo_to_tf_saved_model_test.py)
 add_stablehlo_python_test(stablehlo-python-vhlo vhlo.py)
 
+if(STABLEHLO_ENABLE_PYTHON_TF_TESTS)
+  add_stablehlo_python_test(stablehlo-python-stablehlo-to-saved-model stablehlo_to_tf_saved_model_test.py)
+endif()
+
+
 add_dependencies(check-stablehlo-quick check-stablehlo-python)