From 11ce6e329997ff23f61763a346960235e5c8fdc2 Mon Sep 17 00:00:00 2001 From: diyarora Date: Sun, 28 Jun 2026 13:11:11 +0000 Subject: [PATCH] Fix Parameter Misalignment --- ...symmetric_input_square_kernel__strided_padded__grouped.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/KernelBench/level1/73_conv_transposed_3D_asymmetric_input_square_kernel__strided_padded__grouped.py b/KernelBench/level1/73_conv_transposed_3D_asymmetric_input_square_kernel__strided_padded__grouped.py index d8634831624d94..ab08d67792fd08 100644 --- a/KernelBench/level1/73_conv_transposed_3D_asymmetric_input_square_kernel__strided_padded__grouped.py +++ b/KernelBench/level1/73_conv_transposed_3D_asymmetric_input_square_kernel__strided_padded__grouped.py @@ -17,7 +17,7 @@ class Model(nn.Module): """ def __init__(self, in_channels: int, out_channels: int, kernel_size: int, stride: int = 1, padding: int = 0, output_padding: int = 0, groups: int = 1, bias: bool = False): super(Model, self).__init__() - self.conv_transpose3d = nn.ConvTranspose3d(in_channels, out_channels, kernel_size=(kernel_size, kernel_size, kernel_size), stride=stride, padding=padding, groups=groups, bias=bias) + self.conv_transpose3d = nn.ConvTranspose3d(in_channels, out_channels, kernel_size=(kernel_size, kernel_size, kernel_size), stride=stride, padding=padding, output_padding=output_padding, groups=groups, bias=bias) def forward(self, x: torch.Tensor) -> torch.Tensor: """ @@ -41,6 +41,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: width = 128 stride = 2 padding = 1 +output_padding = 0 groups = 4 def get_inputs(): @@ -48,4 +49,4 @@ def get_inputs(): return [x] def get_init_inputs(): - return [in_channels, out_channels, kernel_size, stride, padding, groups] \ No newline at end of file + return [in_channels, out_channels, kernel_size, stride, padding, output_padding, groups] \ No newline at end of file