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
S390x inductor support #111367
S390x inductor support #111367
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/111367
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 2ab18d7 with merge base 619ae87 (): This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Use arch compile flags. They are needed for vectorization support on s390x. This change fixes multiple tests in test_cpu_repro.py
This change fixes remaining tests in test_cpu_repro.py
Make str return 'zvector' instead of 'vx' since it is used to generate macros like -DCPU_CAPABILITY=ZVECTOR. Rework vectorization detection for s390x. Also rework macros generation to allow more than 1 macro provided.
Provide list of directories instead of -I directives. Function _write_ninja_file_to_build_library is eventually called and it expects list of paths. If include directive is provided, it generates bogus output like: -I/home/user/pytorch/-I/home/user/pytorch/torch/include -I/home/user/pytorch/torch/include/torch/csrc/api/include ... when original input is: -I/home/user/pytorch/torch/include -I/home/user/pytorch/torch/include/torch/csrc/api/include ... Update PCH handling to work with list of paths instead of include directives. This change fixes most tests in test/inductor/test_cpp_wrapper.py on s390x.
dc3479e
to
299a590
Compare
@@ -1048,7 +1064,7 @@ def get_include_and_linking_paths( | |||
vec_isa: VecISA = invalid_vec_isa, | |||
cuda: bool = False, | |||
aot_mode: bool = False, | |||
) -> Tuple[str, str, str, str]: | |||
) -> Tuple[Any, str, str, str, str]: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Isn't this just a str?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
No, it's a list of strings.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
List[str] then, pretty please? :)
@pytorchbot merge |
Merge failedReason: This PR needs a If not, please add the To add a label, you can comment to pytorchbot, for example For more information, see Details for Dev Infra teamRaised by workflow job |
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Follow up to #111367 Pull Request resolved: #111793 Approved by: https://github.com/ezyang, https://github.com/Skylion007
Use arch compile flags. They are needed for vectorization support on s390x. Implement new helper functions for inductor. This change fixes multiple tests in test_cpu_repro.py Pull Request resolved: pytorch#111367 Approved by: https://github.com/ezyang
Follow up to pytorch#111367 Pull Request resolved: pytorch#111793 Approved by: https://github.com/ezyang, https://github.com/Skylion007
Use arch compile flags. They are needed for vectorization support on s390x. Implement new helper functions for inductor. This change fixes multiple tests in test_cpu_repro.py Pull Request resolved: pytorch#111367 Approved by: https://github.com/ezyang
Follow up to pytorch#111367 Pull Request resolved: pytorch#111793 Approved by: https://github.com/ezyang, https://github.com/Skylion007
Follow up to pytorch#111367 Pull Request resolved: pytorch#111793 Approved by: https://github.com/ezyang, https://github.com/Skylion007
Use arch compile flags. They are needed for vectorization support on s390x.
Implement new helper functions for inductor.
This change fixes multiple tests in test_cpu_repro.py
cc @jgong5 @mingfeima @XiaobingSuper @sanchitintel @ashokei @jingxu10 @voznesenskym @penguinwu @EikanWang @Guobing-Chen @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @aakhundov @ColinPeppler