search

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.42k stars 3.4k forks source link

[Bug] nn.Conv3D groups not working correctly #17062

Open DeclK opened 1 month ago

DeclK commented 1 month ago

nn.Conv3D is bugged https://github.com/apache/tvm/blob/78a1f80bf24f1a1114f2ed7d17563d267bb38cc9/python/tvm/relax/frontend/nn/modules.py#L314-L325

line 325 should use in_channels instead of self.in_channels to make groups work. Otherwise in_channels calculated at line 317 would not be effectively used in weight parameters.