-
Notifications
You must be signed in to change notification settings - Fork 486
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.
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
Add dynamo torch.Tensor.new
test.
#6661
Conversation
This PR should only be merged after: |
c465d5a
to
c3bc422
Compare
c3bc422
to
52fa414
Compare
52fa414
to
3f30293
Compare
I'm moving this one forward, since #6659 will take some time to get merged. |
@JackCaoG Could you take a look at it whenever you have some time? |
change lgtm but somehow CI doesn't get trigger. Can you do another push to the branch to see if it will trigger the CI? Otherwise start a new pr might be better. |
@JackCaoG Is this ok to land? All the CI test are passing, now. |
def foo(x): | ||
return x.new(*x.size()) + x | ||
|
||
optfoo = torch.compile(backend="openxla")(foo) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lol is it a new way of calling compile? it used to be torch.compile(foo, backend="openxla")
This PR adds a test for #6009.
The actual fix is implemented in the PyTorch main repository: pytorch/pytorch#121075
@miladm @JackCaoG