PaddleAPITest/tester/__init__.py at 79012220163dfed0d451ac86828ee21a612e0454 · tjujingzong/PaddleAPITest · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# tester/__init__.py

from typing import TYPE_CHECKING, Any

__all__ = [
    'APITestBase',
    'APITestAccuracy',
    'APITestPaddleOnly',
    'APITestCINNVSDygraph',
    'APITestPaddleGPUPerformance',
    'APITestTorchGPUPerformance',
    'APITestPaddleTorchGPUPerformance',
    'APITestAccuracyStable',
    'APITestCustomDeviceVSCPU',
    'APITestPaddleDeviceVSGPU',
    'paddle_to_torch',
    'TensorConfig',
    'APIConfig',
    'analyse_configs',
    'USE_CACHED_NUMPY',
    'cached_numpy',
    'get_cfg',
    'set_cfg'
]

if TYPE_CHECKING:
    from .base import APITestBase
    from .accuracy import APITestAccuracy
    from .paddle_only import APITestPaddleOnly
    from .paddle_gpu_performance import APITestPaddleGPUPerformance
    from .torch_gpu_performance import APITestTorchGPUPerformance
    from .paddle_torch_gpu_performance import APITestPaddleTorchGPUPerformance
    from .paddle_cinn_vs_dygraph import APITestCINNVSDygraph
    from .accuracy_stable import APITestAccuracyStable
    from .paddle_device_vs_cpu import APITestCustomDeviceVSCPU
    from .paddle_device_vs_gpu import APITestPaddleDeviceVSGPU
    from . import paddle_to_torch
    from .api_config import (
        TensorConfig,
        APIConfig,
        analyse_configs,
        USE_CACHED_NUMPY,
        cached_numpy,
        get_cfg,
        set_cfg
    )

def __getattr__(name: str) -> Any:
    if name not in __all__:
        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

    if name == 'APITestBase':
        from .base import APITestBase
        return APITestBase
    elif name == 'APITestAccuracy':
        from .accuracy import APITestAccuracy
        return APITestAccuracy
    elif name == 'APITestPaddleOnly':
        from .paddle_only import APITestPaddleOnly
        return APITestPaddleOnly
    elif name == 'APITestCINNVSDygraph':
        from .paddle_cinn_vs_dygraph import APITestCINNVSDygraph
        return APITestCINNVSDygraph
    elif name == 'APITestPaddleGPUPerformance':
        from .paddle_gpu_performance import APITestPaddleGPUPerformance
        return APITestPaddleGPUPerformance
    elif name == 'APITestTorchGPUPerformance':
        from .torch_gpu_performance import APITestTorchGPUPerformance
        return APITestTorchGPUPerformance
    elif name == 'APITestPaddleTorchGPUPerformance':
        from .paddle_torch_gpu_performance import APITestPaddleTorchGPUPerformance
        return APITestPaddleTorchGPUPerformance
    elif name == 'APITestAccuracyStable':
        from .accuracy_stable import APITestAccuracyStable
        return APITestAccuracyStable
    elif name == 'APITestCustomDeviceVSCPU':
        from .paddle_device_vs_cpu import APITestCustomDeviceVSCPU
        return APITestCustomDeviceVSCPU
    elif name == 'APITestPaddleDeviceVSGPU':
        from .paddle_device_vs_gpu import APITestPaddleDeviceVSGPU
        return APITestPaddleDeviceVSGPU
    elif name == 'paddle_to_torch':
        from . import paddle_to_torch
        return paddle_to_torch
    elif name == 'TensorConfig':
        from .api_config import TensorConfig
        return TensorConfig
    elif name == 'APIConfig':
        from .api_config import APIConfig
        return APIConfig
    elif name == 'analyse_configs':
        from .api_config import analyse_configs
        return analyse_configs
    elif name == 'USE_CACHED_NUMPY':
        from .api_config import USE_CACHED_NUMPY
        return USE_CACHED_NUMPY
    elif name == 'cached_numpy':
        from .api_config import cached_numpy
        return cached_numpy
    elif name == 'get_cfg':
        from .api_config import get_cfg
        return get_cfg
    elif name == 'set_cfg':
        from .api_config import set_cfg
        return set_cfg

    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")