Expose lazy all number special scalars

pytorch / xla

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

https://pytorch.org/xla

Other

2.49k stars 483 forks source link

Closed rpsilva-aws closed 1 week ago

rpsilva-aws commented 2 weeks ago

Expose the getter for the explicit flag that treats all scalars as special - namely not routing the value to device data.

rpsilva-aws commented 2 weeks ago

@JackCaoG @ManfeiBai Small one to expose this to Python XLA module bindings.