[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(

test/cinn/ops/test_floor_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_gather_nd_op.py

@@ -14,16 +14,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
+import os
 import unittest
+from itertools import product
+
+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 logging
-import os
-from itertools import product
 
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="gather_nd")

test/cinn/ops/test_gather_op.py

@@ -14,17 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
+import os
 import unittest
+from itertools import product
+
+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 *
-import logging
-import os
-from itertools import product
 
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="gather")

test/cinn/ops/test_gaussian_random_op.py

@@ -14,12 +14,13 @@
 # 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 cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_gelu_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(

test/cinn/ops/test_identity_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_is_finite_op.py

@@ -14,13 +14,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import numpy as np
 import unittest
+
+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_is_inf_op.py

@@ -14,13 +14,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import numpy as np
 import unittest
+
+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_is_nan_op.py

@@ -14,13 +14,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import numpy as np
 import unittest
+
+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_isclose_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_left_shift_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_log_op.py

@@ -12,12 +12,13 @@
 # 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
 from op_test_helper import TestCaseHelper
+
 import paddle
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 class TestLogOp(OpTest):

test/cinn/ops/test_logical_and_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_logical_not_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_logical_or_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_logical_right_shift_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_logical_xor_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_lookup_table_op.py

@@ -14,13 +14,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.functional as F
-import cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_matmul_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(

test/cinn/ops/test_max_op.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_min_op.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_mod_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_mul_op.py

@@ -14,15 +14,17 @@
 # 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 paddle.nn.functional as F
-import cinn
-from cinn.frontend import *
-from cinn.common import *
-import sys
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_multiply_op.py

@@ -15,11 +15,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_negative_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_one_hot_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(

test/cinn/ops/test_pool2d_op.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 *
 from paddle import _C_ops
 
 

test/cinn/ops/test_popc_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_pow_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_randint_op.py

@@ -14,14 +14,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from random import seed
 import unittest
+from random import seed
+
+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_reciprocal_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 *
 
 
 class TestReciprocalOp(OpTest):

test/cinn/ops/test_reduce_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 *
 
 paddle.seed(2)
 np.random.seed(2)

test/cinn/ops/test_reduce_op_new.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_relu6_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_relu_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.nn.functional as F
-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
+import paddle.nn.functional as F
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_remainder_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_repeat_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.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_reshape_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_resize_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 *
 from paddle.vision.transforms import functional as F
 
 # paddle resize is based on cv2 module

test/cinn/ops/test_reverse_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_right_shift_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_round_op.py

@@ -12,12 +12,13 @@
 # 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 cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_rsqrt_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_scale_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_scatter_add.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, run_test
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_scatter_assign_op.py

@@ -15,12 +15,13 @@
 # limitations under the License.
 
 import numpy as np
-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_select_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_sigmoid_op.py

@@ -14,13 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.nn.functional as F
-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
+import paddle.nn.functional as F
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_sign_op.py

@@ -15,12 +15,13 @@
 # limitations under the License.
 
 import numpy as np
-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_sin_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_sinh_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_slice_assign_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.frontend import *
 from cinn.common import *
+from cinn.frontend import *
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 def paddle_slice_assign(data, update, axes, starts, ends, strides):
     assert len(axes) == len(starts) == len(ends) == len(strides)

test/cinn/ops/test_slice_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_softmax_op.py

@@ -12,13 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.nn.functional as F
-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
+import paddle.nn.functional as F
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_sort_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
 from op_test_helper import TestCaseHelper, run_test
 
+import paddle
+
 
 class TestSortOp(OpTest):
     def setUp(self):

test/cinn/ops/test_split_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_sqrt_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_squeeze_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
 from op_test_helper import TestCaseHelper
 
+import paddle
+
 
 class TestSqueezeOp(OpTest):
     def setUp(self):

test/cinn/ops/test_subtract_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_sum_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_tan_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_tanh_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_top_k_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, run_test
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."

test/cinn/ops/test_transpose_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_triangular_solve_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, run_test
 
+import paddle
+
 
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "triangular solve op support GPU only now."

test/cinn/ops/test_trunc_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_unary_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_uniform_random_op.py

@@ -14,12 +14,13 @@
 # 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 cinn
-from cinn.frontend import *
-from cinn.common import *
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_zero_dim_tensor.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 *
 
 
 def cinn_dtype_convert(dtype_str):

test/cinn/passes/pass_test.py

@@ -12,12 +12,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.frontend import NetBuilder, Variable
-from cinn.frontend import get_default_program_pass, get_default_graph_pass
 import logging
 import os
 from test.cinn.ops.op_test import OpTest
 
+from cinn.frontend import (
+    NetBuilder,
+    Variable,
+    get_default_graph_pass,
+    get_default_program_pass,
+)
+
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="pass_test")
 

test/cinn/passes/test_auto_cast_pass.py

@@ -15,9 +15,10 @@
 # limitations under the License.
 
 import unittest
-from pass_test import PassTest
-from cinn.frontend import *
+
 from cinn.common import *
+from cinn.frontend import *
+from pass_test import PassTest
 
 
 class TestAutoCastPass(PassTest):

test/cinn/passes/test_expand_zero_dim_pass.py

@@ -15,10 +15,11 @@
 # limitations under the License.
 
 import unittest
-from pass_test import PassTest
-from cinn.frontend import *
-from cinn.common import *
+
 import numpy as np
+from cinn.common import *
+from cinn.frontend import *
+from pass_test import PassTest
 
 
 class TestExpandZeroDimPass(PassTest):

test/cinn/passes/test_transpose_floding_input_pass.py

@@ -15,9 +15,10 @@
 # limitations under the License.
 
 import unittest
-from pass_test import PassTest
-from cinn.frontend import *
+
 from cinn.common import *
+from cinn.frontend import *
+from pass_test import PassTest
 
 
 class TestTransposeFoldingInputPass(PassTest):

test/cinn/passes/test_transpose_floding_output_pass.py

@@ -15,9 +15,10 @@
 # limitations under the License.
 
 import unittest
-from pass_test import PassTest
-from cinn.frontend import *
+
 from cinn.common import *
+from cinn.frontend import *
+from pass_test import PassTest
 
 
 class TestTransposeFoldingOutputPass(PassTest):

test/cinn/pool_utils.py

@@ -15,9 +15,10 @@
 # limitations under the License.
 
 import math
-import numpy as np
 import sys
 
+import numpy as np
+
 
 def pool2d(np_data, attrs, dtype="float32"):
     pool_type = "max"

test/cinn/test_common.py

@@ -15,11 +15,9 @@
 # limitations under the License.
 
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn import Target
+from cinn import Target, ir, lang, runtime
 from cinn.common import *
 
 

test/cinn/test_computation.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.fluid as fluid
-import paddle.static as static
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle
+import paddle.fluid as fluid
+import paddle.static as static
 
 assert len(sys.argv) == 3
 enable_gpu = sys.argv.pop()

test/cinn/test_efficientnet.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_facedet.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_frontend.py

@@ -14,19 +14,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle
+import paddle.fluid as fluid
 
 assert len(sys.argv) == 1 + 2 + 1  # model and enable_gpu count
 enable_gpu = sys.argv.pop()

test/cinn/test_hlir_framework.py

@@ -12,9 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.framework import *
 import unittest
+
 import numpy as np
+from cinn.framework import *
 
 
 class TensorTest(unittest.TestCase):

test/cinn/test_ir.py

@@ -15,14 +15,12 @@
 # limitations under the License.
 
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.optim import *
-from cinn import Target
+from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.ir import *
+from cinn.optim import *
 
 
 class TestIR(unittest.TestCase):

test/cinn/test_matmul.py

@@ -15,13 +15,10 @@
 # limitations under the License.
 
 import unittest
-import numpy as np
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn import Target
-from cinn import utils
+import numpy as np
+from cinn import Target, ir, lang, runtime, utils
 from cinn.poly import create_stages
 
 

test/cinn/test_mobilenetv1.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_mobilenetv2.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_netbuilder.py

@@ -14,19 +14,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.static as static
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle
+import paddle.static as static
 
 enable_gpu = sys.argv.pop()
 

test/cinn/test_op_benchmark.py

@@ -14,19 +14,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle
-import paddle.static as static
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle
+import paddle.static as static
 
 assert len(sys.argv) == 2
 enable_gpu = sys.argv.pop()

test/cinn/test_op_broadcast.py

@@ -14,18 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
+import logging
 import math
-import numpy as np
+import unittest
+
 import cinn
-from cinn import frontend
-from cinn import runtime
-from cinn import lang
-from cinn import framework
-from cinn import ir
-from cinn import common
+import numpy as np
+from cinn import common, framework, frontend, ir, lang, runtime
 from cinn.poly import create_stages
-import logging
 from test_utils import SingleOpTester
 
 

test/cinn/test_op_nn.py

@@ -14,22 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import numpy as np
-import unittest
+import logging
 import math
+import unittest
+
 import cinn
-from cinn import frontend
-from cinn import runtime
-from cinn import lang
-from cinn import framework
-from cinn import ir
-from cinn import common
+import conv2d_utils
+import numpy as np
+import pool_utils
+from cinn import common, framework, frontend, ir, lang, runtime
 from cinn.poly import create_stages
-import logging
 from test_utils import SingleOpTester
-import pool_utils
-import conv2d_utils
+
+import paddle as paddle
 
 
 class OpTest_relu(SingleOpTester):

test/cinn/test_op_transform.py

@@ -14,22 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
+import logging
 import math
-import numpy as np
+import os
+import unittest
+
 import cinn
-from cinn import frontend
-from cinn import runtime
-from cinn import lang
-from cinn import framework
-from cinn import ir
-from cinn import common
+import numpy as np
+from cinn import common, framework, frontend, ir, lang, runtime
 from cinn.poly import create_stages
-import logging
 from test_utils import SingleOpTester
+
 import paddle
 import paddle.static as static
-import os
 
 os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
 

test/cinn/test_packed_func.py

@@ -15,12 +15,12 @@
 # limitations under the License.
 
 import unittest
-import numpy as np
-import cinn
-from cinn import ir
-from cinn import CINNValue
 from math import isclose
 
+import cinn
+import numpy as np
+from cinn import CINNValue, ir
+
 
 class TestPackedFunc(unittest.TestCase):
     def setUp(self):

test/cinn/test_paddle_model_convertor.py

@@ -14,18 +14,20 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os, sys
-import numpy as np
+import argparse
+import logging
+import os
+import sys
 import unittest
-import logging, argparse
-
-import paddle
 
+import numpy as np
+from cinn.common import DefaultNVGPUTarget, is_compiled_with_cuda
 from cinn.frontend import PaddleModelConvertor
-from cinn.common import is_compiled_with_cuda, DefaultNVGPUTarget
 from cinn.runtime import seed as cinn_seed
-from ops.op_test import OpTestTool
 from op_mappers.op_mapper_test import OpMapperTest
+from ops.op_test import OpTestTool
+
+import paddle
 
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="paddle_model_convertor")

test/cinn/test_pe_elementwise.py

@@ -15,16 +15,13 @@
 # limitations under the License.
 
 import unittest
+
 import cinn
 import numpy as np
-from cinn import runtime
-from cinn import ir
-from cinn.poly import create_stages
-from cinn import lang
-from cinn import Target
-from cinn import pe
-from cinn.common import *
 import scipy
+from cinn import Target, ir, lang, pe, runtime
+from cinn.common import *
+from cinn.poly import create_stages
 from scipy import special
 
 

test/cinn/test_pe_reduction.py

@@ -15,13 +15,10 @@
 # limitations under the License.
 
 import unittest
+
 import cinn
 import numpy as np
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn import Target
-from cinn import pe
+from cinn import Target, ir, lang, pe, runtime
 from cinn.common import *
 from cinn.poly import create_stages
 

test/cinn/test_pe_transform.py

@@ -15,15 +15,12 @@
 # limitations under the License.
 
 import unittest
+
 import cinn
 import numpy as np
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn import Target
-from cinn import pe
-from cinn.poly import create_stages
+from cinn import Target, ir, lang, pe, runtime
 from cinn.common import *
+from cinn.poly import create_stages
 
 
 class TestPETransform(unittest.TestCase):

test/cinn/test_resnet.py

@@ -14,19 +14,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_resnet18.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_resnet50.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_squeezenet.py

@@ -14,20 +14,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle as paddle
-import paddle.fluid as fluid
-from cinn.frontend import *
-from cinn import Target
-from cinn.framework import *
+import sys
+import time
 import unittest
+
 import cinn
-from cinn import runtime
-from cinn import ir
-from cinn import lang
-from cinn.common import *
 import numpy as np
-import sys
-import time
+from cinn import Target, ir, lang, runtime
+from cinn.common import *
+from cinn.framework import *
+from cinn.frontend import *
+
+import paddle as paddle
+import paddle.fluid as fluid
 
 enable_gpu = sys.argv.pop()
 model_dir = sys.argv.pop()

test/cinn/test_utils.py

@@ -14,18 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
+import logging
 import math
-import numpy as np
+import unittest
+
 import cinn
-from cinn import frontend
-from cinn import runtime
-from cinn import lang
-from cinn import framework
-from cinn import ir
-from cinn import common
+import numpy as np
+from cinn import common, framework, frontend, ir, lang, runtime
 from cinn.poly import create_stages
-import logging
 
 
 class SingleOpTester(unittest.TestCase):

tools/cinn/gen_c++_tutorial.py

@@ -23,9 +23,9 @@ This script helps to extract the tutorial content from a C++ source file.
 #  - @ROC, the code block inside a C++ multi-line string guard `ROC()ROC`,
 #          display as a markdown code block.
 
+import logging
 import sys
 from typing import List
-import logging
 
 
 class Markdown(object):

tools/cinn/paddle_benchmark/paddle_save_model.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

tools/cinn/paddle_benchmark/paddle_test_benchmark.py

@@ -14,10 +14,11 @@
 
 import argparse
 import time
+
 import numpy as np
-from paddle.fluid.core import AnalysisConfig
-from paddle.fluid.core import create_paddle_predictor
+
 import paddle.inference as paddle_infer
+from paddle.fluid.core import AnalysisConfig, create_paddle_predictor
 
 
 def main():

tools/cinn/paddle_benchmark/test_paddle_ops.py

@@ -12,15 +12,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import sys, os
+import argparse
+import os
+import sys
+import time
+
 import numpy as np
+
 import paddle
 import paddle.fluid as fluid
 import paddle.static as static
-import time
-import argparse
-from paddle.fluid.core import AnalysisConfig
-from paddle.fluid.core import create_paddle_predictor
+from paddle.fluid.core import AnalysisConfig, create_paddle_predictor
 
 
 def set_config(op_name, input_shapes, enable_gpu=False):

tools/cinn/tvm_benchmark/test_topi_default.py

@@ -12,14 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
+import timeit
+
+import numpy
 import tvm
 import tvm.testing
-from tvm import te
-import numpy
-import timeit
-from tvm.contrib import tar, ndk
-import os
-from tvm import topi
+from tvm import te, topi
+from tvm.contrib import ndk, tar
 
 dtype = ["float32", "float32", "float32", "float32"]
 target = "llvm"

tools/cinn/tvm_benchmark/tvm_graph_with_single_op.py

@@ -15,15 +15,12 @@
 import os
 
 import numpy as np
-
 import tvm
-from tvm import te
-from tvm import autotvm
-from tvm import relay
+import tvm.contrib.graph_runtime as runtime
 import tvm.relay.testing
-from tvm.autotvm.tuner import XGBTuner, GATuner, RandomTuner, GridSearchTuner
+from tvm import autotvm, relay, te
+from tvm.autotvm.tuner import GATuner, GridSearchTuner, RandomTuner, XGBTuner
 from tvm.contrib.utils import tempdir
-import tvm.contrib.graph_runtime as runtime
 
 # To test different ops, change this single-op network.
 # See https://github.com/apache/incubator-tvm/blob/main/docs/langref/relay_op.rst to get the op list.