Manfei
Manfei
This is the step three PR: Add multi while_loop situation test case father PR: (child of) https://github.com/pytorch/xla/pull/6603 source PR: https://github.com/pytorch/xla/pull/6527
fix https://github.com/pytorch/xla/issues/6090
## 🚀 Feature OpenXLA proposed to move TSL to XLA, and there would be change in PyTorch/XLA to include this moving: change `@tsl` to `@xla/tsl` ## Motivation OpenXLA proposal: https://github.com/openxla/community/pull/100...
Due to `torch.while_loop` is defined in [torch/_higher_order_ops/while_loop.py](https://github.com/pytorch/pytorch/blob/abadbbc4b03138ebe5a892e19ae06eaa41b104d0/torch/_higher_order_ops/while_loop.py#L46), not described in [native_functions.yaml](https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/native_functions.yaml), lower `torch.while_loop` via python dispatch like [PR#5763](https://github.com/pytorch/xla/pull/5763) this is the source PR for all other PRs other PRs: https://github.com/pytorch/xla/pull/6532,...