# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
"""
GPU-generic schedule rules.
For CUDA/ROCm/Vulkan/Metal-specific rules, use `tvm.dlight.cuda/rocm/vulkan/metal` instead
"""

from .fallback import Fallback  # noqa: F401
from .element_wise import ElementWise  # noqa: F401
from .gemv import GEMV  # noqa: F401
from .gemv_dequantize import GEMVWithDequantizeInfo  # noqa: F401
from .general_reduction import GeneralReduction  # noqa: F401
from .matmul import (
    Matmul,  # noqa: F401
    MatmulTensorizationMMA,  # noqa: F401
    MatmulTensorizationWMMA,  # noqa: F401
    MatmulTensorizationMFMA,  # noqa: F401
)
from .matmul_mma_dequantize import (
    MatmulTensorizationMMAWithDequantizeInfo,  # noqa: F401
)
from .matmul_wmma import MatmulTensorizationLegacy  # noqa: F401

from .reduction import Reduction  # noqa: F401
from .transpose import Transpose  # noqa: F401