New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Implement empty_like in torch aten ops #8029

Merged

ManfeiBai merged 1 commit into pytorch:master from nupurbaghel:fix-div

Sep 17, 2024

Contributor

nupurbaghel commented Sep 17, 2024

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)


          implement empty_like in torch aten ops

c772122

ManfeiBai approved these changes

View reviewed changes

Collaborator

ManfeiBai left a comment

Thanks, LGTM

ManfeiBai merged commit 0da723f into pytorch:master

3 checks passed

ManfeiBai added 2024Q3-manfei-review torchxla2 torchxla2-op labels

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

2024Q3-manfei-review torchxla2-op torchxla2