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.

Sign up for GitHub

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

Jump to bottom

[autoparallel] add experimental permute handler #2029

Merged

feifeibear merged 3 commits into hpcaitech:main from YuliangLiu0306:feature/add_experimental_permute_handler

Nov 27, 2022

Contributor

YuliangLiu0306 commented Nov 25, 2022

What does this PR do

This PR integrates torch.reshape, torch.Tensor.reshape into view handler, and add related runtime passes for them.
Complete the permute handler for torch.permute, torch.Tensor.permute, and transpose handler for torch.transpose, torch.Tensor.transpose.
Fix a bug in _all_gahter, which leads to unreported bugs in non-contiguous tensors communication.
Add unit test to test the numerical correctness for above handlers.

YuliangLiu0306 added 3 commits

November 25, 2022 16:54


          finish permute handler

de1ce1d


          [autoparallel] add permute and spose handler

4ec9a4c


          integrate transpose test into permute test

096e634

YuliangLiu0306 added the Run Build and Test label

feifeibear approved these changes

View reviewed changes

feifeibear merged commit 81330b0 into hpcaitech:main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment