diff --git a/torch/_inductor/kernel/conv.py b/torch/_inductor/kernel/conv.py index 37a760a90e1e..44aef074457a 100644 --- a/torch/_inductor/kernel/conv.py +++ b/torch/_inductor/kernel/conv.py @@ -5,6 +5,7 @@ import logging from typing import cast, List, Optional, Sequence, Tuple, TYPE_CHECKING, TypedDict import torch + from .. import config, ir from ..lowering import ( @@ -245,11 +246,11 @@ def conv_layout( ir.ir_node_to_tensor(x, guard_shape=True), ir.ir_node_to_tensor(weight, guard_shape=True), ir.ir_node_to_tensor(bias, guard_shape=True), - stride, - tuple(V.graph.sizevars.size_hint(p) for p in padding), # type: ignore[arg-type] + V.graph.sizevars.size_hints(stride), # type: ignore[arg-type] + V.graph.sizevars.size_hints(padding), # type: ignore[arg-type] dilation, transposed, - tuple(V.graph.sizevars.size_hint(p) for p in output_padding), # type: ignore[arg-type] + V.graph.sizevars.size_hints(output_padding), # type: ignore[arg-type] groups, ) sizes = ir.convert_shape_to_inductor(output.size())