Update on "[inductor] weight prepack for _convolution_transpose_pointwise"

chunyuan-w · chunyuan-w · commit 7ccfae31ff26 · 2022-12-19T15:42:24.000Z
This PR implements weight prepack for `_convolution_transpose_pointwise`, similar to #88988. In addition, we add a kernel for Meta tensor input to reduce the compilation time. cc VitalyFedyunin jgong5 mingfeima XiaobingSuper sanchitintel ashokei jingxu10 mlazos soumith voznesenskym yanboliang penguinwu anijain2305 EikanWang Guobing-Chen zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 desertfire [ghstack-poisoned]
diff --git a/aten/src/ATen/native/mkldnn/Conv.cpp b/aten/src/ATen/native/mkldnn/Conv.cpp
@@ -677,7 +677,7 @@ Tensor _mkldnn_convolution_transpose(
 
   if (bias.defined()) {
     const ideep::tensor b = itensor_from_tensor(bias);
-    ideep::convolution_transpose_forward::compute_v2(
+    ideep::convolution_transpose_forward::compute(
         x,
         w,
         b,
@@ -688,14 +688,9 @@ Tensor _mkldnn_convolution_transpose(
         padding_r(padding, output_padding),
         dilation.vec(),
         groups,
-        ideep::scale_t(),
-        ideep::scale_t(),
-        ideep::scale_t(),
-        ideep::zero_point_t(),
-        ideep::zero_point_t(),
         op_attr);
   } else {
-    ideep::convolution_transpose_forward::compute_v2(
+    ideep::convolution_transpose_forward::compute(
         x,
         w,
         output_sizes,
@@ -705,11 +700,6 @@ Tensor _mkldnn_convolution_transpose(
         padding_r(padding, output_padding),
         dilation.vec(),
         groups,
-        ideep::scale_t(),
-        ideep::scale_t(),
-        ideep::scale_t(),
-        ideep::zero_point_t(),
-        ideep::zero_point_t(),
         op_attr);
   }
   if (input.is_mkldnn()) {
