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

Update update_torch_ods.sh to use new torch_mlir python layout

pull/304/head
Ramiro Leal-Cavazos 2021-09-22 16:27:48 +00:00 committed by Sean Silva
parent 6d8e7f1bb1
commit 19feb58ef8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
external/torch-mlir/build_tools/update_torch_ods.sh vendored
View File

@ -2,10 +2,10 @@
# Updates auto-generated ODS files for the `torch` dialect. # Updates auto-generated ODS files for the `torch` dialect.
set -e set -e
src_dir="$(realpath $(dirname $0)/..)" src_dir="$(realpath $(dirname $0)/../../..)"
build_dir="$(realpath "${TORCH_MLIR_BUILD_DIR:-$src_dir/build}")" build_dir="$(realpath "${TORCH_MLIR_BUILD_DIR:-$src_dir/build}")"
torch_ir_dir="${src_dir}/include/torch-mlir/Dialect/Torch/IR" torch_ir_dir="${src_dir}/external/torch-mlir/include/torch-mlir/Dialect/Torch/IR"
python_packages_dir="${build_dir}/tools/torch-mlir/python_packages" python_packages_dir="${build_dir}/python_packages"
#ninja -C "${build_dir}" #ninja -C "${build_dir}"
PYTHONPATH="${python_packages_dir}/torch_mlir" python \ PYTHONPATH="${python_packages_dir}/torch_mlir" python \