Merge pull request #667 from InfiniTensor/issue/666

issue/666 - Standardized test imports

Author: PanZezhong1725

test/infinicore/framework/__init__.py

@@ -18,9 +18,10 @@
     infinicore_tensor_from_torch,
     rearrange_tensor,
     convert_infinicore_to_torch,
-    is_integer_dtype,
+    is_broadcast,
     is_complex_dtype,
     is_floating_dtype,
+    is_integer_dtype,
 )
 
 __all__ = [
@@ -51,9 +52,10 @@
     "to_torch_dtype",
     "torch_device_map",
     # Type checking utilities
-    "is_integer_dtype",
+    "is_broadcast",
     "is_complex_dtype",
     "is_floating_dtype",
+    "is_integer_dtype",
     # Benchmarking utilities
     "BenchmarkUtils",
     "BenchmarkResult",

test/infinicore/ops/abs.py

@@ -5,8 +5,7 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
+from framework import BaseOperatorTest, TensorSpec, TestCase, GenericTestRunner
 
 # Test cases format: (in_shape, in_strides_or_None)
 _TEST_CASES_DATA = [

test/infinicore/ops/acos.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # =======================================================================
 # Test cases format: (shape, input_strides_or_None)

test/infinicore/ops/acosh.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # =======================================================================
 # Test cases format: (shape, input_strides_or_None)

test/infinicore/ops/adaptive_avg_pool1d.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # Test cases format: (in_shape, in_strides_or_None, output_size_or_None)
 

test/infinicore/ops/adaptive_avg_pool2d.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # Test cases format: (in_shape, in_strides_or_None, output_size_or_None)
 # adaptive_avg_pool2d maps input HxW to target output size (h, w)

test/infinicore/ops/adaptive_avg_pool3d.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # Test cases format: (in_shape, in_strides_or_None, output_size_or_None)
 # adaptive_avg_pool3d maps input D x H x W to target output size (d, h, w)

test/infinicore/ops/adaptive_max_pool1d.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # Test cases format: (in_shape, in_strides_or_None, output_size_or_None)
 

test/infinicore/ops/adaptive_max_pool2d.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase, TestResult
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 from framework.devices import InfiniDeviceEnum
 
 # Test cases format: (in_shape, in_strides_or_None, output_size_or_None)

test/infinicore/ops/add.py

@@ -5,9 +5,13 @@
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+from framework import (
+    BaseOperatorTest,
+    TensorSpec,
+    TestCase,
+    GenericTestRunner,
+    is_broadcast,
+)
 
 # ==============================================================================
 # Operator-specific configuration