[microNPU] Add support for conv2d running on two cores on U65 #10251
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
ekalda
requested review from
jroesch,
slyubomirsky,
icemelon,
MarisaKirisame,
ZihengJiang,
yzhliu,
vinx13,
mbrookhart,
jwfromm,
zhiics,
anijain2305,
wweic,
junrushao,
comaniac,
tqchen,
areusch and
merrymercy
as code owners
ekalda
force-pushed
the
u65-conv2d_upstream
branch
from
95a2d74
to
84777c4
Compare
ekalda
force-pushed
the
u65-conv2d_upstream
branch
from
26322b4
to
3e4de50
Compare
ekalda
force-pushed
the
u65-conv2d_upstream
branch
from
3e4de50
to
6927f34
Compare
ekalda
force-pushed
the
u65-conv2d_upstream
branch
from
6927f34
to
a3d1858
Compare
The 512 mac variant has two cores that processes the weights in parallel, so we need to split the weights and biases into two and encode them separately. Change-Id: I53791f614288ac4df181b9462fc632d35b934a86
Make the DivideConstants to operate on non-flattened tensors to support two core execution in U65.
ekalda
force-pushed
the
u65-conv2d_upstream
branch
from
a3d1858
to
4d425c2
Compare
shtinsa
pushed a commit
to Deelvin/tvm
that referenced
this pull request
May 17, 2022
…#10251) * [microNPU] Add support for conv2d running on two cores on U65 The 512 mac variant has two cores that processes the weights in parallel, so we need to split the weights and biases into two and encode them separately. Change-Id: I53791f614288ac4df181b9462fc632d35b934a86 * Changes due to rebase * Rebase, improve DivideConstants and expand testing Make the DivideConstants to operate on non-flattened tensors to support two core execution in U65.
SebastianBoblest
pushed a commit
to SebastianBoblest/tvm
that referenced
this pull request
May 27, 2022
…#10251) * [microNPU] Add support for conv2d running on two cores on U65 The 512 mac variant has two cores that processes the weights in parallel, so we need to split the weights and biases into two and encode them separately. Change-Id: I53791f614288ac4df181b9462fc632d35b934a86 * Changes due to rebase * Rebase, improve DivideConstants and expand testing Make the DivideConstants to operate on non-flattened tensors to support two core execution in U65.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
The 512 mac variant has two cores that processes the weights in
parallel, so we need to split the weights and biases into two
and encode them separately.
cc @Mousius