-
Notifications
You must be signed in to change notification settings - Fork 22.2k
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
[jit] Emit script function calls during tracing. #25089
Conversation
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead.
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. ghstack-source-id: a20b41b56c12143743e0842a17adf6ffe159480b Pull Request resolved: #25089
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. ghstack-source-id: 9809a277ec1f601125ef22b8842027253b6eb7aa Pull Request resolved: #25089
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.
Looks good!
@@ -10547,6 +10546,7 @@ def traced_fn(x): | |||
self.assertTrue(len(list(traced_fn.graph.inputs())) == 1) | |||
FileCheck().check("aten::mm").check("aten::add").run(str(traced_fn.graph)) | |||
|
|||
@_tmp_donotuse_dont_inline_everything |
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.
but it says donotuse
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.
:'(
@@ -565,6 +565,8 @@ def graph_diagnostic_info(): | |||
break # Graphs have already diverged | |||
|
|||
if n_mod.kind() == 'prim::Constant' and not (n_mod.mustBeNone() or n_check.mustBeNone()): | |||
if not n_mod.hasAttribute('value'): |
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.
why did this change?
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. ghstack-source-id: 5d5fb8513f35fd2574f795b239ec1d29838a1e70 Pull Request resolved: #25089
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. Differential Revision: [D16987936](https://our.internmc.facebook.com/intern/diff/D16987936)
Previously, when the tracer encountered a scripted function (or method), it inlined the function into the graph. Now, we emit a CallFunction or CallMethod node instead. ghstack-source-id: b4ef95e3f8f408c094d553597332fe4d1b9f4a96 Pull Request resolved: #25089
Stack from ghstack:
Previously, when the tracer encountered a scripted function (or method), it
inlined the function into the graph. Now, we emit a CallFunction or
CallMethod node instead.
Differential Revision: D16987936