Skip to content

[CMSIS-NN] Initial operator support for Mul#9163

Merged
manupak merged 1 commit into
apache:mainfrom
Mousius:cmsis-mul-operator
Oct 1, 2021
Merged

[CMSIS-NN] Initial operator support for Mul#9163
manupak merged 1 commit into
apache:mainfrom
Mousius:cmsis-mul-operator

Conversation

@Mousius

@Mousius Mousius commented Sep 30, 2021

Copy link
Copy Markdown
Member

This is largely as it says on the tin, it adds Mul support to CMSIS-NN

@asparkhi asparkhi 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.

Thanks Chris. LGTM!

Comment thread src/relay/backend/contrib/cmsisnn/relay_to_tir.cc
Comment thread src/relay/backend/contrib/cmsisnn/relay_to_tir.cc

@manupak manupak 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.

Broadly looks good : one nit and a question.

Comment thread src/relay/backend/contrib/cmsisnn/relay_to_tir.cc
Comment thread src/relay/backend/contrib/cmsisnn/relay_to_tir.cc

@manupak manupak 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. btw, do we have a tracking issue for note down follow ups ?

@Mousius

Mousius commented Sep 30, 2021

Copy link
Copy Markdown
Member Author

@manupa-arm no need, just did it: #9167

This is largely as it says on the tin, it adds Mul support to CMSIS-NN
@Mousius Mousius force-pushed the cmsis-mul-operator branch from 762d7e8 to ab20bbc Compare October 1, 2021 07:07
@manupak manupak self-assigned this Oct 1, 2021

@manupak manupak 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.

Thanks @Mousius!. Lets keep it open for few hours for other timezones and then proceed to merge, as it quite similar to how other operators are added.

@manupak manupak merged commit 6d120c0 into apache:main Oct 1, 2021
@manupak

manupak commented Oct 1, 2021

Copy link
Copy Markdown
Contributor

Thanks @Mousius @ashutoshparkhi.

from tvm import relay


def skip_if_no_reference_system(func):

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.

should we improve this a bit and make it less brittle? it seems like if we ever use this from a different test (e.g. TOPI tests run on GPU) we may need to broaden it a bit. i'm not sure Corstone300 provides a good way to detect it from TVM--but perhaps you guys know better? we could just add a pytest mark and flag --enable-corstone300-tests and add the flag to the relevant tests/scripts/task_*.sh file.

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.

ylc pushed a commit to ylc/tvm that referenced this pull request Jan 7, 2022
This is largely as it says on the tin, it adds Mul support to CMSIS-NN
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
This is largely as it says on the tin, it adds Mul support to CMSIS-NN
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants