search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.5k stars 482 forks source link

Remove unnecessary functions and print statements (#7495) #8352

Closed simonteozw closed 3 weeks ago

simonteozw commented 3 weeks ago

Discovered aten::eye was unnecessary when working on issue #7495

ManfeiBai commented 3 weeks ago

Thanks, why we notice its unnecessary?

Thanks, saw the reason in chat: basically matrix_power failed because aten_eye called arange an incorrect number of times.