-
Notifications
You must be signed in to change notification settings - Fork 487
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 a DynamoSyncInputExecuteTime
counter
#6813
Conversation
test/dynamo/test_dynamo.py
Outdated
@@ -124,6 +124,7 @@ def test_simple_model(self): | |||
xla_y3 = xla_y * 3 | |||
res_xla_dynamo_3 = fn_simple_dynamo(xla_xy, xla_y3) | |||
res_cpu_3 = self.fn_simple(x + y, y * 3) | |||
breakpoint() |
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.
nit: remove 😄
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
LGTM, the |
Let me rebase this pr |
748e713
to
e3a7c37
Compare
opps, @wonjoolee95 do you mind stamping this pr? |
Aot-autograd might apply reshape on the input before passing to
openxla
backend, in that case we need to execute to sync the input. Add a counter to track that.