llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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.
pull/1295/head snapshot-20220829.580
powderluv 2022-08-28 19:11:40 -07:00 committed by GitHub
parent f245613b71
commit c0630da678
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/buildAndTest.yml vendored
View File

@ -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

2
CMakeLists.txt
View File

@ -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)

2
setup.py
View File

@ -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)