pytorch
d61cd03b - Inductor cpp wrapper: support ConvTranspose and fix Convolution ir (#103308)

Commit
1 year ago
Inductor cpp wrapper: support ConvTranspose and fix Convolution ir (#103308) The changes in this PR include: - Support ConvTranspose in cpp wrapper - Fix cpp wrapper support for aten convolution when bias is `not None`: bias is in `args` instead of `kwargs` when it is `not None`. The change is covered by ConvTranspose dynamic shapes UT since we'll fall back to aten convolution in dynamic shape cases. - Fix cpp wrapper support for `inf`. This is a UT added in https://github.com/pytorch/pytorch/issues/101865. The cpp wrapper UT is covered in `test_conv2d_unary` of `test_cpp_wrapper.py`. It's in `slowTest` category and seems not captured in the CI of that PR. I will submit another PR to remove the hard-coded schema in these `ExternKernel`s. Pull Request resolved: https://github.com/pytorch/pytorch/pull/103308 Approved by: https://github.com/jgong5, https://github.com/desertfire
Author
Committer
Parents
Loading