35 lines
916 B
Python
35 lines
916 B
Python
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||
|
|
"""Helion integration for vLLM."""
|
||
|
|
|
||
|
|
import vllm.kernels.helion.ops # noqa: F401 Auto-register all Helion ops
|
||
|
|
from vllm.kernels.helion.config_manager import (
|
||
|
|
ConfigManager,
|
||
|
|
ConfigSet,
|
||
|
|
)
|
||
|
|
from vllm.kernels.helion.register import (
|
||
|
|
ConfiguredHelionKernel,
|
||
|
|
HelionKernelWrapper,
|
||
|
|
get_kernel_by_name,
|
||
|
|
get_registered_kernels,
|
||
|
|
register_kernel,
|
||
|
|
vllm_helion_lib,
|
||
|
|
)
|
||
|
|
from vllm.kernels.helion.utils import canonicalize_gpu_name, get_canonical_gpu_name
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
# Config management
|
||
|
|
"ConfigManager",
|
||
|
|
"ConfigSet",
|
||
|
|
# Kernel registration
|
||
|
|
"ConfiguredHelionKernel",
|
||
|
|
"HelionKernelWrapper",
|
||
|
|
"get_kernel_by_name",
|
||
|
|
"get_registered_kernels",
|
||
|
|
"register_kernel",
|
||
|
|
"vllm_helion_lib",
|
||
|
|
# Utilities
|
||
|
|
"canonicalize_gpu_name",
|
||
|
|
"get_canonical_gpu_name",
|
||
|
|
]
|