[CodeStyle][CINN] format import region in python code via isort (#54963)

Co-authored-by: NSigureMo <sigure.qaq@gmail.com>

paddle/cinn/backends/llvm/generate_runtime_llvm_ir.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import sys
 import subprocess
+import sys
 
 
 def main():

pyproject.toml

@@ -18,9 +18,6 @@ extend_skip_glob = [
     # see .flake8 for more details
     "python/paddle/fluid/**",
     "python/paddle/utils/gast/**",
-    # Temporarily ignore CINN files, it will fix later
-    "python/cinn/**",
-    "test/cinn/**",
 ]
 
 [tool.ruff]

python/cinn/backends.py

@@ -12,5 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .core_api.backends import ExecutionOptions
-from .core_api.backends import ExecutionEngine
+from .core_api.backends import ExecutionEngine, ExecutionOptions

python/cinn/common.py

@@ -12,9 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .core_api.common import make_const
-from .core_api.common import Target
-from .core_api.common import Type
-from .core_api.common import CINNValue
-from .core_api.common import Void, Int, UInt, Float, Bool, String
 from .core_api.common import *
+from .core_api.common import (
+    Bool,
+    CINNValue,
+    Float,
+    Int,
+    String,
+    Target,
+    Type,
+    UInt,
+    Void,
+    make_const,
+)

python/cinn/lang.py

@@ -12,7 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .core_api.lang import Buffer
-from .core_api.lang import lower
-from .core_api.lang import compute
 from .core_api.lang import *
+from .core_api.lang import Buffer, compute, lower

python/cinn/optim.py

@@ -12,5 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .core_api.optim import simplify
-from .core_api.optim import ir_copy
+from .core_api.optim import ir_copy, simplify

test/cinn/auto_schedule/cost_model/test_cost_model.py

@@ -12,13 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import cinn
-
 import os
 import shutil
-import numpy as np
 import unittest
 
+import cinn
+import numpy as np
 from cinn.auto_schedule.cost_model import CostModel
 
 

test/cinn/conv2d_utils.py

@@ -14,9 +14,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import numpy as np
+
 import paddle
 import paddle.static as static
-import numpy as np
 
 
 def conv2d_native(inputs_data, input_shape, filter_size, attrs, is_depthwise):

test/cinn/fake_model/naive_mul.py

@@ -12,9 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
+import sys
+
 import numpy
-import sys, os
 import numpy as np
+
 import paddle
 import paddle.fluid as fluid
 import paddle.static as static

test/cinn/fake_model/naive_multi_fc.py

@@ -14,9 +14,12 @@
 """
 A fake model with multiple FC layers to test CINN on a more complex model.
 """
+import os
+import sys
+
 import numpy
-import sys, os
 import numpy as np
+
 import paddle
 import paddle.fluid as fluid
 import paddle.static as static

test/cinn/fake_model/resnet_model.py

@@ -12,10 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
+import sys
+
 import numpy
-import paddle
-import sys, os
 import numpy as np
+
+import paddle
 import paddle.fluid as fluid
 import paddle.static as static
 

test/cinn/fusion/test_cast_broadcast_reduce_max.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import unittest
+
 from fusion_test import FusionTest
 
 

test/cinn/fusion/test_reduce_cast.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import unittest
+
 from fusion_test import FusionTest
 
 

test/cinn/fusion/test_select_reduce.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import unittest
+
 from fusion_test import FusionTest
 
 

test/cinn/op_mappers/op_mapper_test.py

@@ -14,19 +14,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from ast import arg
-import os
 import logging
+import os
+import sys
+from ast import arg
 
-import paddle
-from paddle.static import Variable as PaddleVariable
-from paddle.fluid.layer_helper import LayerHelper
-
-from cinn.frontend import NetBuilder, PaddleModelConvertor
 from cinn.common import is_compiled_with_cuda
 from cinn.framework import Scope
+from cinn.frontend import NetBuilder, PaddleModelConvertor
 
-import sys
+import paddle
+from paddle.fluid.layer_helper import LayerHelper
+from paddle.static import Variable as PaddleVariable
 
 sys.path.append("/work/dev_CINN/build/python/tests")
 from test.cinn.ops.op_test import OpTest, OpTestTool

test/cinn/op_mappers/test_argmax_op.py

@@ -14,11 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 import unittest
-from op_mapper_test import OpMapperTest
-from cinn.frontend import *
+
 from cinn.common import *
+from cinn.frontend import *
+from op_mapper_test import OpMapperTest
+
+import paddle
 
 
 class TestArgmaxOp(OpMapperTest):

test/cinn/op_mappers/test_argmin_op.py

@@ -14,11 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 import unittest
-from op_mapper_test import OpMapperTest
-from cinn.frontend import *
+
 from cinn.common import *
+from cinn.frontend import *
+from op_mapper_test import OpMapperTest
+
+import paddle
 
 
 class TestArgminOp(OpMapperTest):

test/cinn/op_mappers/test_argsort_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_assign_value_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_atan2_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_batch_norm_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_bitwise_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_cholesky_op.py

@@ -13,8 +13,10 @@
 # limitations under the License.
 
 import unittest
-from op_mapper_test import OpMapperTest
+
 import numpy as np
+from op_mapper_test import OpMapperTest
+
 import paddle
 
 

test/cinn/op_mappers/test_clip_op.py

@@ -14,11 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 import random
 import unittest
+
 from op_mapper_test import OpMapperTest
 
+import paddle
+
 
 class TestClipOp(OpMapperTest):
     def init_input_data(self):

test/cinn/op_mappers/test_compare_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_conv2d_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_cumsum_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_elementwise_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_expand_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_expand_v2_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_fill_constant_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_flip_op.py

@@ -15,7 +15,9 @@
 # limitations under the License.
 
 import unittest
+
 from op_mapper_test import OpMapperTest
+
 import paddle
 
 

test/cinn/op_mappers/test_gather_nd_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_gather_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_gaussian_random_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_layer_norm_op.py

@@ -14,11 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import functools
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
-import functools
 
 
 class TestLayerNormOp(OpMapperTest):

test/cinn/op_mappers/test_log1p_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_logical_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_lookup_table_op.py

@@ -13,8 +13,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_mul_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_norm_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_one_hot_op.py

@@ -13,8 +13,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_pool2d_op.py

@@ -13,9 +13,11 @@
 # limitations under the License.
 
 import unittest
-import paddle
-from op_mapper_test import OpMapperTest
+
 from cinn.common import *
+from op_mapper_test import OpMapperTest
+
+import paddle
 
 
 @unittest.skipIf(

test/cinn/op_mappers/test_pow_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_randint_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_reduce_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_reverse_op.py

@@ -15,7 +15,9 @@
 # limitations under the License.
 
 import unittest
+
 from op_mapper_test import OpMapperTest
+
 import paddle
 
 

test/cinn/op_mappers/test_roll_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_scale_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_scatter_op.py

@@ -14,12 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import numpy as np
-import paddle
 import random
 import unittest
+
+import numpy as np
 from op_mapper_test import OpMapperTest
 
+import paddle
+
 
 class TestScatterOp(OpMapperTest):
     def init_input_data(self):

test/cinn/op_mappers/test_sign_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_split_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_squeeze_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_stack_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_strided_slice_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_take_along_axis_op.py

@@ -13,8 +13,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest
+
 import paddle
 
 

test/cinn/op_mappers/test_tile_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_transpose2_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_triangular_solve_op.py

@@ -15,7 +15,9 @@
 # limitations under the License.
 
 import unittest
+
 from op_mapper_test import OpMapperTest
+
 import paddle
 
 

test/cinn/op_mappers/test_unary_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_uniform_random_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/op_mappers/test_where_op.py

@@ -15,8 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import numpy as np
 from op_mapper_test import OpMapperTest, logger
+
 import paddle
 
 

test/cinn/ops/op_test.py

@@ -12,18 +12,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
+import os
+import struct
 import unittest
+from contextlib import contextmanager
+
+import numpy as np
 from cinn import Target
-from cinn.frontend import *
 from cinn.common import *
+from cinn.frontend import *
 from cinn.runtime import seed as cinn_seed
-import numpy as np
-import paddle
-import logging
-from contextlib import contextmanager
-import os
 
-import struct
+import paddle
 
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="op_test")

test/cinn/ops/op_test_helper.py

@@ -12,14 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import sys
 import argparse
 import itertools
-import unittest
 import re
-
+import sys
+import unittest
+from typing import List, Union
 from unittest import suite
-from typing import Union, List
 
 parser = argparse.ArgumentParser(description="Argparse for op test helper")
 parser.add_argument(

test/cinn/ops/test_abs_op.py

@@ -15,11 +15,13 @@
 # limitations under the License.
 
 import unittest
+
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_acos_op.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_acosh_op.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_add_op.py

@@ -13,12 +13,13 @@
 # limitations under the License.
 
 import numpy as np
-import paddle
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_arange_op.py

@@ -14,12 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-from cinn.frontend import *
 from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_argsort_op.py

@@ -15,13 +15,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
+
 import paddle
 import paddle.nn.functional as F
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_asin_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_asinh_op.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_atan2_op.py

@@ -12,11 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_atan_op.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_atanh_op.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_batch_norm_op.py

@@ -14,14 +14,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest, sys
+import sys
+import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_binary_elementwise_op.py

@@ -15,12 +15,14 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_bitcast_convert_op.py

@@ -13,14 +13,15 @@
 # limitations under the License.
 
 import unittest
-import numpy as np
-from op_test import OpTest, OpTestTool
-import paddle
+from struct import pack, unpack
+
 import cinn
-from cinn.frontend import *
+import numpy as np
 from cinn.common import *
+from cinn.frontend import *
+from op_test import OpTest, OpTestTool
 
-from struct import pack, unpack
+import paddle
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_bitwise_op.py

@@ -13,11 +13,12 @@
 # limitations under the License.
 
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_broadcast_to_op.py

@@ -14,14 +14,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sys
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
-import sys
 
 
 class TestBroadcastToOp(OpTest):

test/cinn/ops/test_broadcast_to_op_new.py

@@ -14,11 +14,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_cast_op.py

@@ -15,13 +15,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_cbrt_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_ceil_op.py

@@ -14,12 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_cholesky_op.py

@@ -15,12 +15,13 @@
 # limitations under the License.
 
 import numpy as np
-import paddle
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_clz_op.py

@@ -15,12 +15,13 @@
 # limitations under the License.
 
 import numpy as np
-import paddle
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 INT32_MAX = (1 << 31) - 1
 INT32_MIN = -(1 << 31)
 INT64_MAX = (1 << 63) - 1

test/cinn/ops/test_comparison_op.py

@@ -12,12 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-from cinn.frontend import *
 from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_concat_op.py

@@ -14,12 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_constant_op.py

@@ -14,12 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_conv2d_op.py

@@ -13,17 +13,18 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import cinn
-import numpy as np
-import paddle
 import unittest
 
-from cinn.frontend import *
+import cinn
+import numpy as np
 from cinn.common import *
+from cinn.frontend import *
 from cinn.runtime import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 set_cinn_cudnn_deterministic(True)
 paddle.fluid.set_flags({'FLAGS_cudnn_deterministic': 1})
 

test/cinn/ops/test_cos_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_cosh_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_depthwise_conv2d_op.py

@@ -12,13 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import cinn
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
 import paddle.nn as nn
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_divide_op.py

@@ -14,12 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-from cinn.frontend import *
 from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_dropout_infer_op.py

@@ -13,13 +13,15 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_erf_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_exp_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_expand_dims.py

@@ -14,11 +14,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_fill_constant_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_floor_divide_op.py

@@ -15,14 +15,16 @@
 # limitations under the License.
 
 import unittest
+
+import cinn
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
+
 import paddle
 import paddle.nn.functional as F
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(