Skip to content

[BYOC] FTVMAnnotateTarget method signature update - #6786

Merged
comaniac merged 1 commit into
apache:mainfrom
d-smirnov:register_op_attr
Nov 4, 2020
Merged

[BYOC] FTVMAnnotateTarget method signature update#6786
comaniac merged 1 commit into
apache:mainfrom
d-smirnov:register_op_attr

Conversation

@d-smirnov

Copy link
Copy Markdown
Contributor

This PR changes signature of FTVMAnnotateTarget to runtime::TypedPackedFunc<bool(const Expr& expr)>
which allows to utilise extra information (e.g. an operation output type) from passed expr argument.

@comaniac comaniac left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.
cc @zhiics @masahi
Also cc @trevor-m @mbaret. Please help double check.

@mbaret mbaret left a comment

Copy link
Copy Markdown
Contributor

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.

Comment thread tests/python/contrib/test_arm_compute_lib/test_conv2d.py Outdated
Comment thread python/tvm/relay/op/contrib/arm_compute_lib.py
Comment thread python/tvm/relay/op/contrib/arm_compute_lib.py Outdated

@mbaret mbaret left a comment

Copy link
Copy Markdown
Contributor

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.

Comment thread python/tvm/relay/op/contrib/arm_compute_lib.py Outdated
Comment thread python/tvm/relay/op/contrib/arm_compute_lib.py Outdated
@trevor-m

Copy link
Copy Markdown
Contributor

tensorrt.py changes look good

@comaniac comaniac left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@comaniac comaniac left a comment

Copy link
Copy Markdown
Contributor

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.

Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)>
which allows to utilise extra information from passed expr argument.
@comaniac

comaniac commented Nov 3, 2020

Copy link
Copy Markdown
Contributor

@mbaret PTAL and approve and request changes explicitly. Thanks.

@mbaret mbaret left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm now

@comaniac
comaniac merged commit 6e36fc4 into apache:main Nov 4, 2020
@comaniac

comaniac commented Nov 4, 2020

Copy link
Copy Markdown
Contributor

Thanks @d-smirnov @mbaret @lhutton1 @trevor-m

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)>
which allows to utilise extra information from passed expr argument.
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)>
which allows to utilise extra information from passed expr argument.
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
Signature of FTVMAnnotateTarget changed to runtime::TypedPackedFunc<bool(const Expr& expr)>
which allows to utilise extra information from passed expr argument.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants