Update

[ghstack-poisoned]
pytorch · ZhiweiYan-96 · Nov 12, 2024 · Nov 12, 2024 · Nov 12, 2024 · Nov 13, 2024
commit 8b87482bd90a798418e71a00500b23a03347ea9b
diff --git a/aten/src/ATen/native/mkldnn/xpu/Linear.h b/aten/src/ATen/native/mkldnn/xpu/Linear.h
@@ -8,21 +8,6 @@ struct LinearConverter {
     is_fused_ = false;
   }
 
-  // linear with post-ops
-  template <typename Func>
-  Tensor& call(
-      Tensor& result,
-      const Tensor& input,
-      const Tensor& weight,
-      const c10::optional<Tensor>& bias,
-      Func func) {
-    onednn::Attr attr = func();
-    Tensor _bias = bias.has_value() ? bias.value() : at::Tensor();
-    Tensor _input = input.dim() <= 2 ? input : input.contiguous();
-    return impl::matmul_fusion_variants(
-        result, _input, weight, false, attr, is_fused_, _bias);
-  }
-
   Tensor& call(
       Tensor& result,
       const Tensor& input,
@@ -32,7 +17,7 @@ struct LinearConverter {
     Tensor _bias = bias.has_value() ? bias.value() : at::Tensor();
     Tensor _input = input.dim() <= 2 ? input : input.contiguous();
     return impl::matmul_fusion_variants(
-        result, _input, weight, /*trans*/ true, attr, is_fused_, _bias);
+        result, _input, weight, /*trans*/ false, attr, is_fused_, _bias);
   }
 
   bool is_fused() {

diff --git a/test/xpu/test_fusion.py b/test/xpu/test_fusion.py
@@ -82,7 +82,7 @@ def forward(self, x):
                     algorithm = pointwise_info.algorithm
                     fused = torch.ops.mkldnn._linear_pointwise(
                         v,
-                        mod.linear.weight.transpose(1, 0),
+                        mod.linear.weight,
                         mod.linear.bias,
                         attr,
                         scalars,
@@ -116,7 +116,7 @@ def forward(self, x, other):
                 fused = torch.ops.mkldnn._linear_pointwise(
                     input,
                     other,
-                    model.linear.weight.transpose(1, 0),
+                    model.linear.weight,
                     model.linear.bias,
                     attr,
                 )