XLA downcast for S64 and U64 for Neuron

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

https://pytorch.org/xla

Other

2.5k stars 482 forks source link

Closed aws-nm9 closed 3 weeks ago

aws-nm9 commented 4 weeks ago

This PR downcasts the S64 and U64 datatypes for Neuron devices, to prevent unsupported datatypes from being lowered to the compiler.

jeffhataws commented 4 weeks ago

@qihqi will you look at this one also? it is related to https://github.com/pytorch/xla/pull/8335