-
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
[BYOC] FTVMAnnotateTarget method signature update #6786
Conversation
45b4766
to
475835a
Compare
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.
This change may allow us to simplify some logic for Ethos-N as well. I'll leave that for a follow-up patch though.
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.
Some stuff has become registered when it shouldn't be.
tensorrt.py changes look good |
e424083
to
fe3c184
Compare
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.
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.
I'll block it first due to the unresolved comments. @d-smirnov please address the concern raised by @mbaret.
fe3c184
to
87333e0
Compare
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)> which allows to utilise extra information from passed expr argument.
87333e0
to
6e4375b
Compare
@mbaret PTAL and approve and request changes explicitly. Thanks. |
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 now
Thanks @d-smirnov @mbaret @lhutton1 @trevor-m |
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)> which allows to utilise extra information from passed expr argument.
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)> which allows to utilise extra information from passed expr argument.
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)> which allows to utilise extra information from passed expr argument.
This PR changes signature of
FTVMAnnotateTarget
toruntime::TypedPackedFunc<bool(const Expr& expr)>
which allows to utilise extra information (e.g. an operation output type) from passed expr argument.