| # Copyright (c) Microsoft Corporation. | |
| # Licensed under the MIT License. | |
| """Base infra""" | |
| from .analysis import ( | |
| BlockInfo, # noqa: F401 | |
| IterInfo, # noqa: F401 | |
| collect_block_iter_vars_used_in_access_region, # noqa: F401 | |
| collect_vars_used_in_prim_expr, # noqa: F401 | |
| detect_dominant_read, # noqa: F401 | |
| is_broadcast_epilogue, # noqa: F401 | |
| normalize_prim_func, # noqa: F401 | |
| ) # noqa: F401 | |
| from .common_schedules import get_block, get_output_blocks, try_inline, try_inline_contiguous_spatial # noqa: F401 | |
| from .base_scheduler import simplify_prim_func # noqa: F401 | |
| from .schedule_rule import ScheduleRule # noqa: F401 | |
| from .tuner import fast_tune, fast_tune_with_dynamic_range # noqa: F401 | |
| from .roller import * | |
| from .arch import CUDA, CDNA # noqa: F401 | |
| from .operator_common import TransformKind, OptimizeStrategy, BackendKind # noqa: F401 | |