From c0630da678ea015ca8757340633c654d5539adac Mon Sep 17 00:00:00 2001 From: powderluv Date: Sun, 28 Aug 2022 19:11:40 -0700 Subject: [PATCH] Disable LTC by default until upstream revert relands (#1303) * Disable LTC by default until upstream revert relands Tracked with the WIP https://github.com/llvm/torch-mlir/pull/1292 * Disable LTC e2e tests temporarily * Update setup.py Disable LTC in setup.py temporarily until upstream is fixed. --- .github/workflows/buildAndTest.yml | 3 ++- CMakeLists.txt | 2 +- setup.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/buildAndTest.yml b/.github/workflows/buildAndTest.yml index 1af6cdc0f..84cec511a 100644 --- a/.github/workflows/buildAndTest.yml +++ b/.github/workflows/buildAndTest.yml @@ -183,4 +183,5 @@ jobs: if: ${{ matrix.os-arch == 'ubuntu-x86_64' && matrix.llvm-build == 'in-tree' }} run: | export PYTHONPATH="$GITHUB_WORKSPACE/build/tools/torch-mlir/python_packages/torch_mlir" - python -m e2e_testing.torchscript.main --config=lazy_tensor_core -v + echo "LTC tests disabled temporarily. https://github.com/llvm/torch-mlir/pull/1292" + # python -m e2e_testing.torchscript.main --config=lazy_tensor_core -v diff --git a/CMakeLists.txt b/CMakeLists.txt index 1a07e7b10..bf031b772 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -49,7 +49,7 @@ if(TORCH_MLIR_ENABLE_MHLO) endif() endif() -option(TORCH_MLIR_ENABLE_LTC "Enables LTC backend" ON) +option(TORCH_MLIR_ENABLE_LTC "Enables LTC backend" OFF) if(TORCH_MLIR_ENABLE_LTC) set(ENV{TORCH_MLIR_ENABLE_LTC} 1) diff --git a/setup.py b/setup.py index 53a0346d9..929c27e8a 100644 --- a/setup.py +++ b/setup.py @@ -81,7 +81,7 @@ class CMakeBuild(build_py): f"-DCMAKE_VISIBILITY_INLINES_HIDDEN=ON", f"-DCMAKE_C_VISIBILITY_PRESET=hidden", f"-DCMAKE_CXX_VISIBILITY_PRESET=hidden", - f"-DTORCH_MLIR_ENABLE_LTC={'ON' if int(os.environ.get('TORCH_MLIR_ENABLE_LTC', 1)) else 'OFF'}", + f"-DTORCH_MLIR_ENABLE_LTC={'OFF' if int(os.environ.get('TORCH_MLIR_ENABLE_LTC', 1)) else 'OFF'}", ] os.makedirs(cmake_build_dir, exist_ok=True)