Disable lazy backend tests if LTC is disabled

llvm · Aug 12, 2022 · 09a605f · 09a605f
1 parent b1276ea
commit 09a605f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -51,6 +51,12 @@ endif()
 
 option(TORCH_MLIR_ENABLE_LTC "Enables LTC backend" ON)
 
+if(TORCH_MLIR_ENABLE_LTC)
+    set(ENV{TORCH_MLIR_ENABLE_LTC} 1)
+else()
+    set(ENV{TORCH_MLIR_ENABLE_LTC} 0)
+endif()
+
 torch_mlir_add_llvm_external_project(
   torch-mlir-dialects
   TORCH_MLIR_DIALECTS

diff --git a/python/test/lit.cfg.py b/python/test/lit.cfg.py
@@ -51,6 +51,9 @@
 # directories.
 config.excludes = ['lit.cfg.py', 'Inputs', 'Examples', 'CMakeLists.txt', 'README.txt', 'LICENSE.txt']
 
+if bool(int(os.environ.get("TORCH_MLIR_ENABLE_LTC", 0))):
+   config.excludes.append("lazy_backend")
+
 # test_source_root: The root path where tests are located.
 config.test_source_root = os.path.dirname(__file__)
 

diff --git a/python/torch_mlir/csrc/base_lazy_backend/utils/string_utils.h b/python/torch_mlir/csrc/base_lazy_backend/utils/string_utils.h
@@ -12,6 +12,7 @@ std::ostream& string_join(std::ostream& out, const std::vector<T>& v, const std:
         if ((i++) > 0) { out << delimiter; }
         out << e;
     }
+    return out;
 }
 
 template <typename T>