[inductor] Check if n is the input tensor of conv_pointwise

ghstack-source-id: 9ef21de7f23b99449b82b5b4d2b2c6e2da641512 Pull Request resolved: #125119
pytorch · Apr 28, 2024 · 12a212b · 12a212b
1 parent 94b328e
commit 12a212b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_inductor/graph.py b/torch/_inductor/graph.py
@@ -1286,7 +1286,7 @@ def debug(msg):
                                 ir.get_stride_order(n.meta["val"].stride()),
                                 allow_padding=True,
                             )
-                        if user.target in need_fixed_channels_last_layout:
+                        if user.target in need_fixed_channels_last_layout and n is user.args[0]:
                             result = ir.ExternKernel.require_stride_order(
                                 result,
                                 ir.get_stride_order(