check stride padding of conv1d

FlagOpen · Nov 11, 2024 · 4f9b9e9 · 4f9b9e9
1 parent d41b935
commit 4f9b9e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/flag_gems/ops/conv2d.py b/src/flag_gems/ops/conv2d.py
@@ -371,12 +371,12 @@ def forward(ctx, input, weight, bias, stride, padding, dilation, groups):
             bias is None or weight.shape[0] == bias.shape[0]
         ), "Incompatible weights ({weight.shape}) and bias ({bias.shape}) shape"
 
-        if isinstance(stride, list):
+        if isinstance(stride, (list, tuple)):
             stride_height, stride_width = stride
         else:
             stride_height = stride_width = stride
 
-        if isinstance(padding, list):
+        if isinstance(padding, (list, tuple)):
             padding_height, padding_width = padding
         else:
             padding_height = padding_width = padding