Skip to content
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

[topi]fix group conv3d pack kernel shape error #12523

Merged
merged 1 commit into from
Aug 22, 2022
Merged

[topi]fix group conv3d pack kernel shape error #12523

merged 1 commit into from
Aug 22, 2022

Conversation

chengven027
Copy link
Contributor

@chengven027 chengven027 commented Aug 22, 2022

Related PR: #12500
hi,
I am sorry i made a mistake about group_conv3d.
There is three case about caculate in_channel // groups // ic_bn.

  1. if in_channel // groups // ic_bn == 0, It means that we only need to tile to 1
  2. if out = in_channel // groups // ic_bn != 0, but the out * ic_bn * group < in_channel . It means that we need to tile to out +1
  3. if out = in_channel // groups // ic_bn != 0, but the out * ic_bn * group == in_channel . It means that we need to tile to out

In the last PR, I didn`t consider the second case. So this PR I fix it.

CC: @masahi

@masahi masahi merged commit e9aad35 into apache:main Aug 22, 2022
xinetzone pushed a commit to daobook/tvm that referenced this pull request Nov 25, 2022
@chengven027 chengven027 deleted the group_conv3d.fix branch January 12, 2024 11:51
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.

2 participants