File size: 900 Bytes
9823a7e | 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 | # 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
|