search

pytorch / xla

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

Implement empty_like in torch aten ops #8028

Closed nupurbaghel closed 2 months ago

nupurbaghel commented 2 months ago

This op is not part of the skip list so no test count changes. Fixes #7386

Did a quick verification by comparing outputs from torch and jax:

$a=torch.empty((2,3), dtype=torch.int32)
$print(torch.empty_like(a))
tensor([[88391536,    32347, 88390000],
        [   32347, 88391088,    32347]], dtype=torch.int32)

$b=jnp.empty((2,3), dtype=jnp.int32)
$jnp.empty_like(b)
Array([[0, 0, 0],
       [0, 0, 0]], dtype=int32)