[CodeStyle][CINN] ruff F401,F632,F811 in test/cinn (#55013)

pyproject.toml

@@ -103,8 +103,5 @@ ignore = [
     "F403",
 ]
 "test/cinn/**" = [
-    "F401",
     "F403",
-    "F632",
-    "F811",
 ]

test/cinn/auto_schedule/cost_model/test_cost_model.py

@@ -16,7 +16,6 @@ import os
 import shutil
 import unittest
 
-import cinn
 import numpy as np
 from cinn.auto_schedule.cost_model import CostModel
 

test/cinn/fake_model/naive_mul.py

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

test/cinn/fake_model/naive_multi_fc.py

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

test/cinn/fake_model/resnet_model.py

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

test/cinn/op_mappers/op_mapper_test.py

@@ -17,18 +17,17 @@
 import logging
 import os
 import sys
-from ast import arg
 
 from cinn.common import is_compiled_with_cuda
 from cinn.framework import Scope
-from cinn.frontend import NetBuilder, PaddleModelConvertor
+from cinn.frontend import PaddleModelConvertor
 
 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
+from test.cinn.ops.op_test import OpTest
 
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="op_test")

test/cinn/op_mappers/test_argsort_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_assign_value_op.py

@@ -17,9 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
-
-import paddle
+from op_mapper_test import OpMapperTest
 
 
 class TestAssignValueOp(OpMapperTest):

test/cinn/op_mappers/test_atan2_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_batch_norm_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_bitwise_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_compare_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_conv2d_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_cumsum_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_elementwise_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_expand_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_expand_v2_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_fill_constant_op.py

@@ -17,7 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_gather_nd_op.py

@@ -17,7 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_gather_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_gaussian_random_op.py

@@ -16,10 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
-
-import paddle
+from op_mapper_test import OpMapperTest
 
 
 class TestGaussianRandomOp(OpMapperTest):

test/cinn/op_mappers/test_layer_norm_op.py

@@ -14,11 +14,9 @@
 # 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
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_log1p_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_logical_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_lookup_table_op.py

@@ -14,8 +14,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_mul_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_norm_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_one_hot_op.py

@@ -14,8 +14,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_pow_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_randint_op.py

@@ -16,10 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
-
-import paddle
+from op_mapper_test import OpMapperTest
 
 
 class TestRandIntOp(OpMapperTest):

test/cinn/op_mappers/test_reduce_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_roll_op.py

@@ -17,7 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_scale_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_sign_op.py

@@ -17,7 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_split_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_squeeze_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_stack_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_strided_slice_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_tile_op.py

@@ -17,7 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_transpose2_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_unary_op.py

@@ -17,7 +17,7 @@
 import unittest
 
 import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/op_mappers/test_uniform_random_op.py

@@ -16,10 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
-
-import paddle
+from op_mapper_test import OpMapperTest
 
 
 class TestUniformRandomOp(OpMapperTest):

test/cinn/op_mappers/test_where_op.py

@@ -16,8 +16,7 @@
 
 import unittest
 
-import numpy as np
-from op_mapper_test import OpMapperTest, logger
+from op_mapper_test import OpMapperTest
 
 import paddle
 

test/cinn/ops/op_test.py

@@ -16,10 +16,8 @@ import logging
 import os
 import struct
 import unittest
-from contextlib import contextmanager
 
 import numpy as np
-from cinn import Target
 from cinn.common import *
 from cinn.frontend import *
 from cinn.runtime import seed as cinn_seed

test/cinn/ops/test_abs_op.py

@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_acos_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_acosh_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_asin_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_asinh_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_atan_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_atanh_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_batch_norm_op.py

@@ -14,11 +14,7 @@
 # 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

test/cinn/ops/test_binary_elementwise_op.py

@@ -16,7 +16,6 @@
 
 import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_bitcast_convert_op.py

@@ -15,7 +15,6 @@
 import unittest
 from struct import pack, unpack
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_broadcast_to_op.py

@@ -14,14 +14,12 @@
 # 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
+from op_test import OpTest
 
 import paddle
 

test/cinn/ops/test_cast_op.py

@@ -14,10 +14,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_conv2d_op.py

@@ -13,10 +13,7 @@
 # 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 unittest
 
-import cinn
-import numpy as np
 from cinn.common import *
 from cinn.frontend import *
 from cinn.runtime import *

test/cinn/ops/test_cos_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_cosh_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_depthwise_conv2d_op.py

@@ -12,7 +12,6 @@
 # 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

test/cinn/ops/test_divide_op.py

@@ -45,7 +45,7 @@ class TestDivOp(OpTest):
         )
 
         # paddle.divide does not support zero division
-        if self.case["y_dtype"] is "int32" or self.case["y_dtype"] is "int64":
+        if self.case["y_dtype"] == "int32" or self.case["y_dtype"] == "int64":
             self.y_np[self.y_np == 0] = 1
 
     def build_paddle_program(self, target):

test/cinn/ops/test_dropout_infer_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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
@@ -40,9 +37,9 @@ class TestDropoutInferOp(OpTest):
         self.x_np = self.random(
             shape=self.case["x_shape"], dtype=self.case["x_dtype"]
         )
-        if self.case["mode"] is 'upscale_in_train':
+        if self.case["mode"] == 'upscale_in_train':
             self.case["cinn_mode"] = 'upscale_in_train'
-        elif self.case["mode"] is 'downscale_in_infer':
+        elif self.case["mode"] == 'downscale_in_infer':
             self.case["cinn_mode"] = 'downgrade_in_infer'
         else:
             raise f"Unknown mode for dropout_infer: {self.case['mode']}"

test/cinn/ops/test_erf_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_exp_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_floor_divide_op.py

@@ -14,17 +14,13 @@
 # See the License for the specific language governing permissions and
 # 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
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_floor_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_gather_nd_op.py

@@ -19,7 +19,6 @@ import os
 import unittest
 from itertools import product
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_gather_op.py

@@ -16,10 +16,7 @@
 
 import logging
 import os
-import unittest
-from itertools import product
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_gaussian_random_op.py

@@ -14,7 +14,6 @@
 # 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

test/cinn/ops/test_gelu_op.py

@@ -14,10 +14,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_identity_op.py

@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_is_finite_op.py

@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
 import numpy as np
 from cinn.common import *

test/cinn/ops/test_is_inf_op.py

@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
 import numpy as np
 from cinn.common import *

test/cinn/ops/test_is_nan_op.py

@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
 import numpy as np
 from cinn.common import *

test/cinn/ops/test_isclose_op.py

@@ -14,7 +14,6 @@
 # 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 *

test/cinn/ops/test_left_shift_op.py

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_log_op.py

@@ -12,7 +12,6 @@
 # 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

test/cinn/ops/test_logical_and_op.py

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_logical_not_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_logical_or_op.py

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_logical_right_shift_op.py

@@ -14,7 +14,6 @@
 # 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 *

test/cinn/ops/test_logical_xor_op.py

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_lookup_table_op.py

@@ -14,7 +14,6 @@
 # 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

test/cinn/ops/test_matmul_op.py

@@ -14,17 +14,13 @@
 # See the License for the specific language governing permissions and
 # 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
 
 
 @OpTestTool.skip_if(

test/cinn/ops/test_mod_op.py

@@ -14,10 +14,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_negative_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_one_hot_op.py

@@ -14,10 +14,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_pow_op.py

@@ -14,7 +14,6 @@
 # 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 *

test/cinn/ops/test_randint_op.py

@@ -15,10 +15,7 @@
 # limitations under the License.
 
 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

test/cinn/ops/test_reciprocal_op.py

@@ -12,13 +12,11 @@
 # See the License for the specific language governing permissions and
 # 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 import OpTest
 from op_test_helper import TestCaseHelper
 
 import paddle

test/cinn/ops/test_reduce_op.py

@@ -16,7 +16,6 @@
 
 import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_reduce_op_new.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_relu6_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_remainder_op.py

@@ -14,7 +14,6 @@
 # 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 *

test/cinn/ops/test_resize_op.py

@@ -14,16 +14,9 @@
 # See the License for the specific language governing permissions and
 # 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
-from paddle.vision.transforms import functional as F
 
 # paddle resize is based on cv2 module
 # This test requires cv2 module (pip3.6 install opencv_python==3.2.0.7)

test/cinn/ops/test_right_shift_op.py

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_round_op.py

@@ -12,7 +12,6 @@
 # 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

test/cinn/ops/test_rsqrt_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_scale_op.py

@@ -12,10 +12,7 @@
 # See the License for the specific language governing permissions and
 # 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

test/cinn/ops/test_sin_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_sinh_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_sqrt_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_tan_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_tanh_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_trunc_op.py

@@ -14,8 +14,6 @@
 # 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

test/cinn/ops/test_unary_elementwise_op.py

@@ -16,7 +16,6 @@
 
 import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/ops/test_uniform_random_op.py

@@ -14,7 +14,6 @@
 # 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

test/cinn/ops/test_zero_dim_tensor.py

@@ -16,7 +16,6 @@
 
 import unittest
 
-import cinn
 import numpy as np
 from cinn.common import *
 from cinn.frontend import *

test/cinn/passes/pass_test.py

@@ -16,12 +16,7 @@ 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,
-)
+from cinn.frontend import NetBuilder, Variable
 
 logging.basicConfig(level=os.environ.get('LOG_LEVEL', 'INFO').upper())
 logger = logging.getLogger(name="pass_test")

test/cinn/test_common.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-import cinn
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 
 

test/cinn/test_computation.py

@@ -17,9 +17,7 @@
 import sys
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *

test/cinn/test_efficientnet.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_facedet.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_frontend.py

@@ -17,14 +17,11 @@
 import sys
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 assert len(sys.argv) == 1 + 2 + 1  # model and enable_gpu count

test/cinn/test_ir.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-import cinn
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.ir import *
 from cinn.optim import *

test/cinn/test_mobilenetv1.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_mobilenetv2.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_netbuilder.py

@@ -17,9 +17,7 @@
 import sys
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *

test/cinn/test_op_benchmark.py

@@ -17,9 +17,7 @@
 import sys
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
@@ -314,7 +312,7 @@ __global__ void fn_conv2d_0_kernel(float* __restrict__ placeholder, float* __res
             "TESTING [mul and add] time cost with shape [128,512]*[256,512]...",
         )
 
-    def atest_matmul(self):
+    def atest_matmul3(self):
         prog = Program()
         a = Variable("A").set_type(Float(32)).set_shape([512, 512])
         b = Variable("B").set_type(Float(32)).set_shape([512, 512])
@@ -440,7 +438,7 @@ typedef char int8_t;
             (tensor_data[0] + tensor_data[1]).reshape(-1), result, atol=1e-4
         )
 
-    def atest_elementwise2(self):
+    def atest_elementwise3(self):
         prog = Program()
         a = Variable("A").set_type(Float(32)).set_shape([4, 1024])
         b = Variable("B").set_type(Float(32)).set_shape([4, 1024])

test/cinn/test_op_broadcast.py

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

test/cinn/test_op_nn.py

@@ -14,20 +14,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import logging
 import math
 import unittest
 
-import cinn
 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
+from cinn import framework
 from test_utils import SingleOpTester
 
-import paddle
-
 
 class OpTest_relu(SingleOpTester):
     def create_target_data(self, inputs_data, attrs):

test/cinn/test_op_transform.py

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

test/cinn/test_packed_func.py

@@ -18,8 +18,7 @@ import unittest
 from math import isclose
 
 import cinn
-import numpy as np
-from cinn import CINNValue, ir
+from cinn import ir
 
 
 class TestPackedFunc(unittest.TestCase):

test/cinn/test_paddle_model_convertor.py

@@ -21,15 +21,10 @@ import sys
 import unittest
 
 import numpy as np
-from cinn.common import (
-    DefaultHostTarget,
-    DefaultNVGPUTarget,
-    is_compiled_with_cuda,
-)
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
 from cinn.frontend import PaddleModelConvertor
 from cinn.runtime import seed as cinn_seed
 from op_mappers.op_mapper_test import OpMapperTest
-from ops.op_test import OpTestTool
 
 import paddle
 

test/cinn/test_pe_elementwise.py

@@ -22,7 +22,6 @@ import scipy
 from cinn import Target, ir, lang, pe, runtime
 from cinn.common import *
 from cinn.poly import create_stages
-from scipy import special
 
 
 class TestPEElementwise(unittest.TestCase):

test/cinn/test_resnet.py

@@ -17,14 +17,11 @@
 import sys
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_resnet18.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_resnet50.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_squeezenet.py

@@ -18,14 +18,11 @@ import sys
 import time
 import unittest
 
-import cinn
 import numpy as np
-from cinn import Target, ir, lang, runtime
 from cinn.common import *
 from cinn.framework import *
 from cinn.frontend import *
 
-import paddle
 from paddle import fluid
 
 enable_gpu = sys.argv.pop()

test/cinn/test_utils.py

@@ -15,13 +15,11 @@
 # limitations under the License.
 
 import logging
-import math
 import unittest
 
 import cinn
 import numpy as np
-from cinn import common, framework, frontend, ir, lang, runtime
-from cinn.poly import create_stages
+from cinn import common, framework, ir, lang, runtime
 
 
 class SingleOpTester(unittest.TestCase):