Run stablehlo tests only when feature is enabled.

tenstorrent · Aug 20, 2024 · 8c9e97f · 8c9e97f
1 parent adfade4
commit 8c9e97f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 0 deletions.
diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -55,6 +55,7 @@ jobs:
         -DCMAKE_BUILD_TYPE=${{ matrix.build.build_type }} \
         -DTTMLIR_ENABLE_RUNTIME=${{ matrix.build.enable_runtime }} \
         -DTTMLIR_ENABLE_RUNTIME_TESTS=${{ matrix.build.enable_runtime }} \
+        -DTTMLIR_ENABLE_STABLEHLO=ON \
         -S ${{ steps.strings.outputs.work-dir }}
 
     - name: Build

diff --git a/test/lit.cfg.py b/test/lit.cfg.py
@@ -24,6 +24,10 @@
 
 config.test_format = lit.formats.ShTest(not llvm_config.use_lit_shell)
 
+# Stablehlo tests can be optionally enabled.
+if config.enable_stablehlo:
+    config.available_features.add("stablehlo")
+
 # suffixes: A list of file extensions to treat as test files.
 config.suffixes = [".mlir"]
 

diff --git a/test/lit.site.cfg.py.in b/test/lit.site.cfg.py.in
@@ -6,6 +6,7 @@ config.mlir_obj_dir = "@MLIR_BINARY_DIR@"
 config.enable_bindings_python = @MLIR_ENABLE_BINDINGS_PYTHON@ and "@TTMLIR_ENABLE_BINDINGS_PYTHON@" == "ON"
 config.ttmlir_obj_root = "@TTMLIR_BINARY_DIR@"
 config.llvm_shlib_ext = "@SHLIBEXT@"
+config.enable_stablehlo = "@TTMLIR_ENABLE_STABLEHLO@" and "@TTMLIR_ENABLE_STABLEHLO@" == "ON"
 
 import lit.llvm
 lit.llvm.initialize(lit_config, config)

diff --git a/test/ttmlir/Dialect/TTIR/stablehlo_to_ttir_addop.mlir b/test/ttmlir/Dialect/TTIR/stablehlo_to_ttir_addop.mlir
@@ -1,3 +1,4 @@
+// REQUIRES: stablehlo
 // RUN: ttmlir-opt --convert-stablehlo-to-ttir %s | FileCheck %s
 #any_device = #tt.operand_constraint<dram|l1|scalar|tile|any_device|any_device_tile>
 module @jit_eltwise_add attributes {} {