File size: 879 Bytes
9823a7e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | # 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
|