xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Add dynamo expand test.

Open ysiraichi opened this issue 11 months ago • 2 comments

This PR adds a test for #5837. The fix was introduced in PyTorch main repository (https://github.com/pytorch/pytorch/pull/121007), but we need PyTorch/XLA for actually exercising the (previously) failing test case.

cc @miladm @JackCaoG

ysiraichi avatar Mar 01 '24 15:03 ysiraichi

You can add a .torch_pin to build with your patch PR. Following the instructions here: https://github.com/pytorch/xla/tree/master/torch_patches

Such that we don't need to add any guards. We just need to land the two patches together.

alanwaketan avatar Mar 04 '24 23:03 alanwaketan

Great, looks like all tests pass! Please ping me once your upstream PR is approved. Also, we need to remove the .torch_pin before landing the PR.

alanwaketan avatar Mar 05 '24 19:03 alanwaketan