-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
Don't run non-tvm_op GraphRuntime nodes in Debug Runtime over RPC. #7512
Conversation
* These are filtered out in SetupOpExecs for normal debug runtime operation.
22f28ae
to
5145a28
Compare
Is this for ops that are extern calls? Like calling into MKL? |
@tkonolige no, the GraphRuntime has two different paths to run operators: one for local operators where non-kernel ops could be done fairly inexpensively (I.e. layout transforms maybe? this isn't actually used) and one for RPC operators (where that is not easy). right now tvm_op == "null" (for graph nodes which are inputs) or "tvm_op" (for graph nodes which are operator function execs). |
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.
LGTM
Thanks @tkonolige @areusch the PR has been merged |
…pache#7512) * Don't run non-tvm_op GraphRuntime nodes in Debug Runtime over RPC. * These are filtered out in SetupOpExecs for normal debug runtime operation. * retrigger CI * retrigger CI * address tkonolige comment
…pache#7512) * Don't run non-tvm_op GraphRuntime nodes in Debug Runtime over RPC. * These are filtered out in SetupOpExecs for normal debug runtime operation. * retrigger CI * retrigger CI * address tkonolige comment
@tmoreau89 @tqchen @jroesch @tkonolige