tree 97c921700d913b370454695ed8723cae07d7e01d
parent 834e8f04644f6ed072d69c03a819b701fb9f7950
author Wang, Eikan <eikan.wang@intel.com> 1675391075 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1675568182 +0000

To vectorize long datatype as mask index (#91076)

In this PR, we record the current fx node being executed to cache additional information to simply the vectorization checker. In addition, we supported `masked` in this PR by simplifying it as `mask_load` to support `max_pool2d`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/91076
Approved by: https://github.com/jgong5, https://github.com/desertfire, https://github.com/jansel