fix format

Author: huangyiqun

src/flag_gems/ops/__init__.py

@@ -230,9 +230,6 @@
 )
 from flag_gems.ops.zeros import zeros
 from flag_gems.ops.zeros_like import zeros_like
-from flag_gems.ops.conv1d import conv1d
-from flag_gems.ops.conv2d import conv2d
-from flag_gems.ops.conv3d import conv3d
 
 __all__ = [
     "_conv_depthwise2d",

src/flag_gems/ops/conv1d.py

@@ -14,7 +14,8 @@ def conv1d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1):
         stride_width = stride
 
     if (padding == "same"):
-        assert stride == 1, "Doesn't support any stride values other than 1 in padding = 'same' mode, received stride value {stride}" 
+        assert stride == 1, "Doesn't support any stride values other than 1 \
+            in padding = 'same' mode, received stride value {stride}" 
         il = input.shape[-1]
         kernel_size = weight.shape[-1]
         padding_width = math.ceil((stride * (il - 1) + 1 + dilation * (kernel_size - 1) - il) / 2)

src/flag_gems/ops/conv2d.py

@@ -593,7 +593,8 @@ def backward(ctx, out_grad):
 # todo test SymInt[2] of stride or padding
 def conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1):
     if (padding == "same"):
-        assert stride == 1, "Doesn't support any stride values other than 1 in padding = 'same' mode, received stride value {stride}" 
+        assert stride == 1, "Doesn't support any stride values other than 1 \
+            in padding = 'same' mode, received stride value {stride}" 
         ih = input.shape[-2]
         iw = input.shape[-1]
         kernel_size_h = weight.shape[-2]

src/flag_gems/ops/conv3d.py

@@ -237,11 +237,13 @@ def conv3d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1):
     if isinstance(stride, (list, tuple)):
         stride_depth, stride_height, stride_width = stride
         if (padding == "same"):
-            assert (stride_depth == 1 and stride_height == 1 and stride_width == 1), "Doesn't support any stride values other than 1 in padding = 'same' mode, received stride value {stride}" 
+            assert (stride_depth == 1 and stride_height == 1 and stride_width == 1), "Doesn't \
+                support any stride values other than 1 in padding = 'same' mode, received stride value {stride}" 
     else:
         stride_depth = stride_height = stride_width = stride
         if (padding == "same"):
-            assert stride == 1, "Doesn't support any stride values other than 1 in padding = 'same' mode, received stride value {stride}" 
+            assert stride == 1, "Doesn't support any stride values other than 1 \
+                in padding = 'same' mode, received stride value {stride}" 
 
     if isinstance(dilation, (list, tuple)):
         dilation_depth, dilation_height, dilation_width = dilation

tests/test_convolution_ops.py

@@ -43,6 +43,7 @@ def test_accuracy_conv1d(shape, kernel, stride, padding, dtype):
     if flag_gems.vendor_name == "mthreads" and dtype == torch.float16:
         del os.environ["MUSA_ENABLE_SQMMA"]
 
+
 @pytest.mark.skipif(flag_gems.vendor_name == "kunlunxin", reason="RESULT TODOFIX")
 @pytest.mark.conv1d_padding
 @pytest.mark.parametrize("shape, kernel", SHAPE_CONV1D)
@@ -169,6 +170,7 @@ def test_accuracy_conv2d(shape, kernel, stride, padding, groups, dtype, dilation
     if flag_gems.vendor_name == "mthreads" and dtype == torch.float16:
         del os.environ["MUSA_ENABLE_SQMMA"]
 
+
 @pytest.mark.skipif(flag_gems.vendor_name == "hygon", reason="RESULT TODOFIX")
 @pytest.mark.skipif(flag_gems.vendor_name == "kunlunxin", reason="RESULT TODOFIX")
 @pytest.mark.conv2d_padding