Skip to content

vllm.kernels.helion.ops

Helion kernel implementation.

Importing this package does NOT register any kernels. Runtime code imports the specific op module it needs, e.g.::

from vllm.kernels.helion.ops import scaled_mm  # noqa: F401

which triggers that op's @register_kernel as an import side effect.

Tools that need the full registry (e.g. scripts/autotune_helion_kernels.py) call import_all_ops() to force every op module to register.

Modules:

Functions:

  • import_all_kernels –

    Import every kernel submodule so all @register_kernel decorators run.

import_all_kernels()

Import every kernel submodule so all @register_kernel decorators run.

Returns:

  • list[str] –

    The fully-qualified module names that were imported.

Source code in vllm/kernels/helion/ops/__init__.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
def import_all_kernels() -> list[str]:
    """Import every kernel submodule so all ``@register_kernel`` decorators run.

    Returns:
        The fully-qualified module names that were imported.
    """
    imported: list[str] = []
    for module_info in pkgutil.iter_modules(__path__):
        if module_info.ispkg:
            continue
        module_name = f"{__name__}.{module_info.name}"
        importlib.import_module(module_name)
        imported.append(module_name)
    return imported