Update update_torch_ods.sh to use new torch_mlir python layout

redbopo · Sep 22, 2021 · 19feb58 · 19feb58
1 parent 6d8e7f1
commit 19feb58
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/external/torch-mlir/build_tools/update_torch_ods.sh b/external/torch-mlir/build_tools/update_torch_ods.sh
@@ -2,10 +2,10 @@
 # Updates auto-generated ODS files for the `torch` dialect.
 set -e
 
-src_dir="$(realpath $(dirname $0)/..)"
+src_dir="$(realpath $(dirname $0)/../../..)"
 build_dir="$(realpath "${TORCH_MLIR_BUILD_DIR:-$src_dir/build}")"
-torch_ir_dir="${src_dir}/include/torch-mlir/Dialect/Torch/IR"
-python_packages_dir="${build_dir}/tools/torch-mlir/python_packages"
+torch_ir_dir="${src_dir}/external/torch-mlir/include/torch-mlir/Dialect/Torch/IR"
+python_packages_dir="${build_dir}/python_packages"
 
 #ninja -C "${build_dir}"
 PYTHONPATH="${python_packages_dir}/torch_mlir" python \