Skip to content

[microNPU] Use TF reference kernels for codegen tests where possible#10762

Merged
AndrewZhaoLuo merged 1 commit into
apache:mainfrom
lhutton1:optionally-use-reference-kernels
Mar 28, 2022
Merged

[microNPU] Use TF reference kernels for codegen tests where possible#10762
AndrewZhaoLuo merged 1 commit into
apache:mainfrom
lhutton1:optionally-use-reference-kernels

Conversation

@lhutton1

Copy link
Copy Markdown
Contributor

Uses reference kernels for the codegen tests when the version of Tensorflow is >= 2.5.0 (as this is the first version this functionality was added). Otherwise, fallback to running the tests without.

cc @leandron @manupa-arm @ekalda @NicolaLancellotti

@leandron leandron 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, thanks for this fix @lhutton1!

@NicolaLancellotti NicolaLancellotti 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! Thanks, @lhutton1

Uses reference kernels for the codegen tests when the version of
Tensorflow is >= 2.5.0 (as this is the first version this functionality
was added). Otherwise, fallback to running the tests without.

Change-Id: I92b24ad259d2fda2fed497aa0fe6d7f11a0db85a
@lhutton1 lhutton1 force-pushed the optionally-use-reference-kernels branch from 1182500 to e8a8730 Compare March 24, 2022 19:13
@github-actions github-actions Bot requested review from leandron and manupak March 24, 2022 19:14
@AndrewZhaoLuo AndrewZhaoLuo merged commit 4bd64b8 into apache:main Mar 28, 2022
pfk-beta pushed a commit to pfk-beta/tvm that referenced this pull request Apr 11, 2022
…pache#10762)

Uses reference kernels for the codegen tests when the version of
Tensorflow is >= 2.5.0 (as this is the first version this functionality
was added). Otherwise, fallback to running the tests without.

Change-Id: I92b24ad259d2fda2fed497aa0fe6d7f11a0db85a
@lhutton1 lhutton1 deleted the optionally-use-reference-kernels branch April 12, 2022 10:15
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