Skip to content

Commit cd62d10

Browse files
committed
issue/497 - improved test framework
greatly reduced the code required for specific operators; added strided tensor support;
1 parent 41e1bb2 commit cd62d10

7 files changed

Lines changed: 642 additions & 246 deletions

File tree

test/infinicore/framework/__init__.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,36 @@
1-
from .base import TestConfig, TestRunner, TestCase
1+
from .base import TensorSpec, TestConfig, TestRunner, TestCase, BaseOperatorTest
22
from .utils import (
33
create_infinicore_tensor,
4+
create_strided_infinicore_tensor,
45
compare_results,
56
create_test_comparator,
67
debug,
78
get_tolerance,
89
profile_operation,
910
rearrange_tensor,
1011
convert_infinicore_to_torch,
12+
is_tensor_contiguous,
1113
)
1214
from .config import get_test_devices, get_args
1315
from .devices import InfiniDeviceEnum, InfiniDeviceNames, torch_device_map
1416
from .datatypes import to_torch_dtype, to_infinicore_dtype
1517

1618
__all__ = [
19+
"TensorSpec",
1720
"TestConfig",
1821
"TestRunner",
1922
"TestCase",
23+
"BaseOperatorTest",
2024
"create_infinicore_tensor",
25+
"create_strided_infinicore_tensor",
2126
"compare_results",
2227
"create_test_comparator",
2328
"convert_infinicore_to_torch",
2429
"debug",
2530
"get_tolerance",
2631
"profile_operation",
2732
"rearrange_tensor",
33+
"is_tensor_contiguous",
2834
"get_test_devices",
2935
"get_args",
3036
"InfiniDeviceEnum",

0 commit comments

Comments
 (0)