[CodeStyle][CINN] ruff F403 in test/cinn (#55255)

pyproject.toml

@@ -97,7 +97,3 @@ ignore = [
 "test/dygraph_to_static/test_loop.py" = ["C416", "F821"]
 # Ignore unnecessary lambda in dy2st unittest test_lambda
 "test/dygraph_to_static/test_lambda.py" = ["PLC3002"]
-# Temporarily ignore CINN files, it will fix later
-"test/cinn/**" = [
-    "F403",
-]

test/cinn/op_mappers/test_argmax_op.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.frontend import *
 from op_mapper_test import OpMapperTest
 
 import paddle

test/cinn/op_mappers/test_argmin_op.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.frontend import *
 from op_mapper_test import OpMapperTest
 
 import paddle

test/cinn/op_mappers/test_pool2d_op.py

@@ -14,7 +14,7 @@
 
 import unittest
 
-from cinn.common import *
+from cinn.common import is_compiled_with_cudnn
 from op_mapper_test import OpMapperTest
 
 import paddle

test/cinn/ops/op_test.py

@@ -18,8 +18,17 @@ import struct
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import (
+    BFloat16,
+    Bool,
+    DefaultHostTarget,
+    DefaultNVGPUTarget,
+    Float,
+    Float16,
+    Int,
+    UInt,
+    is_compiled_with_cuda,
+)
 from cinn.runtime import seed as cinn_seed
 
 import paddle

test/cinn/ops/test_abs_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_acos_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_acosh_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_add_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_arange_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_argmax_op.py

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

test/cinn/ops/test_argmin_op.py

@@ -16,8 +16,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_argsort_op.py

@@ -16,8 +16,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_asin_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_asinh_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_atan2_op.py

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

test/cinn/ops/test_atan_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_atanh_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_batch_norm_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_binary_elementwise_op.py

@@ -13,12 +13,11 @@
 # 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 numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/ops/test_bitcast_convert_op.py

@@ -16,8 +16,8 @@ import unittest
 from struct import pack, unpack
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/ops/test_bitwise_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_broadcast_to_op.py

@@ -17,8 +17,8 @@
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import Float
+from cinn.frontend import NetBuilder
 from op_test import OpTest
 
 import paddle

test/cinn/ops/test_broadcast_to_op_new.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_cast_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_cbrt_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_ceil_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_cholesky_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_clz_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_comparison_op.py

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

test/cinn/ops/test_concat_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_constant_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_conv2d_op.py

@@ -14,9 +14,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
-from cinn.runtime import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
+from cinn.runtime import set_cinn_cudnn_deterministic
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_cos_op.py

@@ -13,9 +13,8 @@
 # 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.
-
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_cosh_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_depthwise_conv2d_op.py

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

test/cinn/ops/test_divide_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_dropout_infer_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_erf_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_exp_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_expand_dims.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_fill_constant_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_floor_divide_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_floor_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_gather_nd_op.py

@@ -20,8 +20,8 @@ import unittest
 from itertools import product
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import Int, is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/ops/test_gather_op.py

@@ -18,8 +18,8 @@ import logging
 import os
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import Int, is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_gaussian_random_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_gelu_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_identity_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_is_finite_op.py

@@ -16,8 +16,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_is_inf_op.py

@@ -16,8 +16,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_is_nan_op.py

@@ -16,8 +16,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_isclose_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_left_shift_op.py

@@ -14,8 +14,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_log_op.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.frontend import NetBuilder
 from op_test import OpTest
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_logical_and_op.py

@@ -14,8 +14,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_logical_not_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_logical_or_op.py

@@ -14,8 +14,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_logical_right_shift_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_logical_xor_op.py

@@ -14,8 +14,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_lookup_table_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_matmul_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_max_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_min_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_mod_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_mul_op.py

@@ -13,8 +13,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_multiply_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_negative_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_one_hot_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_pool2d_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cudnn
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_popc_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_pow_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_randint_op.py

@@ -16,8 +16,8 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/ops/test_reciprocal_op.py

@@ -14,8 +14,7 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.frontend import NetBuilder
 from op_test import OpTest
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_reduce_op.py

@@ -17,8 +17,8 @@
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import Bool, Float, Int, is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/ops/test_reduce_op_new.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_relu6_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_relu_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_remainder_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_repeat_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_reshape_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_resize_op.py

@@ -15,9 +15,6 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
-
 # paddle resize is based on cv2 module
 # This test requires cv2 module (pip3.6 install opencv_python==3.2.0.7)
 # @OpTestTool.skip_if(not is_compiled_with_cuda(),

test/cinn/ops/test_reverse_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_right_shift_op.py

@@ -14,8 +14,8 @@
 
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_round_op.py

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

test/cinn/ops/test_rsqrt_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_scale_op.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_scatter_add.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import Float, Int, is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper, run_test
 

test/cinn/ops/test_scatter_assign_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_select_op.py

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

test/cinn/ops/test_sigmoid_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_sign_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_sin_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_sinh_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_slice_assign_op.py

@@ -15,8 +15,8 @@
 # limitations under the License.
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_slice_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_softmax_op.py

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

test/cinn/ops/test_sort_op.py

@@ -14,8 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.frontend import NetBuilder
 from op_test import OpTest
 from op_test_helper import TestCaseHelper, run_test
 

test/cinn/ops/test_split_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_sqrt_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_squeeze_op.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.frontend import NetBuilder
 from op_test import OpTest
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_subtract_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_sum_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_tan_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_tanh_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_top_k_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper, run_test
 

test/cinn/ops/test_transpose_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_triangular_solve_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper, run_test
 

test/cinn/ops/test_trunc_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_unary_elementwise_op.py

@@ -17,8 +17,8 @@
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/ops/test_uniform_random_op.py

@@ -14,8 +14,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 from op_test_helper import TestCaseHelper
 

test/cinn/ops/test_zero_dim_tensor.py

@@ -17,8 +17,8 @@
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
+from cinn.common import Bool, Float, Int, is_compiled_with_cuda
+from cinn.frontend import NetBuilder
 from op_test import OpTest, OpTestTool
 
 import paddle

test/cinn/passes/test_auto_cast_pass.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.frontend import *
 from pass_test import PassTest
 
 

test/cinn/passes/test_expand_zero_dim_pass.py

@@ -17,8 +17,6 @@
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.frontend import *
 from pass_test import PassTest
 
 

test/cinn/passes/test_transpose_floding_input_pass.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.frontend import *
 from pass_test import PassTest
 
 

test/cinn/passes/test_transpose_floding_output_pass.py

@@ -16,8 +16,6 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.frontend import *
 from pass_test import PassTest
 
 

test/cinn/test_common.py

@@ -16,7 +16,7 @@
 
 import unittest
 
-from cinn.common import *
+from cinn.common import Bool, Float, Int, UInt, make_const
 
 
 class TestType(unittest.TestCase):

test/cinn/test_computation.py

@@ -18,9 +18,8 @@ import sys
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget, Float
+from cinn.frontend import Computation, NetBuilder
 
 import paddle
 from paddle import fluid, static

test/cinn/test_efficientnet.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_facedet.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_frontend.py

@@ -18,9 +18,8 @@ import sys
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_hlir_framework.py

@@ -15,7 +15,8 @@
 import unittest
 
 import numpy as np
-from cinn.framework import *
+from cinn import Target
+from cinn.framework import Tensor
 
 
 class TensorTest(unittest.TestCase):

test/cinn/test_ir.py

@@ -16,9 +16,8 @@
 
 import unittest
 
-from cinn.common import *
-from cinn.ir import *
-from cinn.optim import *
+from cinn.ir import Expr, Var
+from cinn.optim import simplify
 
 
 class TestIR(unittest.TestCase):

test/cinn/test_mobilenetv1.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_mobilenetv2.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_netbuilder.py

@@ -18,9 +18,8 @@ import sys
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget, Float
+from cinn.frontend import NetBuilder
 
 import paddle
 from paddle import static

test/cinn/test_op_benchmark.py

@@ -18,9 +18,8 @@ import sys
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget, Float
+from cinn.frontend import Program, Variable
 
 import paddle
 from paddle import static

test/cinn/test_pe_elementwise.py

@@ -20,7 +20,6 @@ import cinn
 import numpy as np
 import scipy
 from cinn import Target, ir, lang, pe, runtime
-from cinn.common import *
 from cinn.poly import create_stages
 
 

test/cinn/test_pe_reduction.py

@@ -19,7 +19,6 @@ import unittest
 import cinn
 import numpy as np
 from cinn import Target, ir, lang, pe, runtime
-from cinn.common import *
 from cinn.poly import create_stages
 
 

test/cinn/test_pe_transform.py

@@ -19,7 +19,6 @@ import unittest
 import cinn
 import numpy as np
 from cinn import Target, ir, lang, pe, runtime
-from cinn.common import *
 from cinn.poly import create_stages
 
 

test/cinn/test_resnet.py

@@ -18,9 +18,8 @@ import sys
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_resnet18.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_resnet50.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid
 

test/cinn/test_squeezenet.py

@@ -19,9 +19,8 @@ import time
 import unittest
 
 import numpy as np
-from cinn.common import *
-from cinn.framework import *
-from cinn.frontend import *
+from cinn.common import DefaultHostTarget, DefaultNVGPUTarget
+from cinn.frontend import Interpreter
 
 from paddle import fluid