-
Notifications
You must be signed in to change notification settings - Fork 48
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
torch.aten.avg_pool3d to linalg #300
Comments
This was actually quite involved to implement. Namely, we cannot apply pooling_ndhwc_sum generically to In That being said, the best approach I thought was just writing out the entire convolution using I thought it would be a bit cumbersome to constantly trying to achieve computational correctness, so I wanted to solve an easier problem. So, I implemented, I implemented the lowering, |
Hi @wu-s-john, are you still working on this op? |
Required in model VideoResNet_vaiq_int8
%1473 = torch.aten.avg_pool3d %1471, %1472, %60, %551, %false, %false, %none : !torch.vtensor<[1,512,14,14,14],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[1,512,1,1,1],f32>
Interested in implement the lowering for
AvgPool3dOp
from Torch to linalg.Seems that there is no native 3D convolution method for doing a sum or max in linalg. I would be happy to implement.
#215
llvm/torch-mlir#3030
The text was updated successfully, but these errors were encountered: