Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

build: update ODS and shape library when updating PyTorch #1450

Merged
merged 1 commit into from
Oct 2, 2022
Merged

build: update ODS and shape library when updating PyTorch #1450

merged 1 commit into from
Oct 2, 2022

Conversation

ashay
Copy link
Collaborator

@ashay ashay commented Oct 2, 2022

Updating the PyTorch version may break the Torch-MLIR build, as it did
recently, since the PyTorch update caused the shape library to change,
but the shape library was not updated in the commit for updating
PyTorch.

This patch introduces a new default-off environment variable to the
build_linux_packages.sh script called TM_UPDATE_ODS_AND_SHAPE_LIB
which instructs the script to run the update_torch_ods.sh and
update_shape_lib.sh scripts.

However, running these scripts requires an in-tree build and the tests
that run for an in-tree build of Torch-MLIR are more comprehensive than
those that run for an out-of-tree build, so this patch also swaps out
the out-of-tree build for an in-tree build.

@powderluv
Copy link
Collaborator

Thank you for this. I think we were hitting some issues related to this downstream in SHARK.

Updating the PyTorch version may break the Torch-MLIR build, as it did
recently, since the PyTorch update caused the shape library to change,
but the shape library was not updated in the commit for updating
PyTorch.

This patch introduces a new default-off environment variable to the
build_linux_packages.sh script called `TM_UPDATE_ODS_AND_SHAPE_LIB`
which instructs the script to run the update_torch_ods.sh and
update_shape_lib.sh scripts.

However, running these scripts requires an in-tree build and the tests
that run for an in-tree build of Torch-MLIR are more comprehensive than
those that run for an out-of-tree build, so this patch also swaps out
the out-of-tree build for an in-tree build.
@ashay
Copy link
Collaborator Author

ashay commented Oct 2, 2022

Rebased with main

@ashay ashay merged commit da02390 into llvm:main Oct 2, 2022
@ashay ashay deleted the ashay/update-shape-libs-in-roll-pytorch branch October 2, 2022 23:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants