Skip to content

vllm ir oot kernel registration#4

Open
wxsIcey wants to merge 3 commits intoluka/vllm-ir/rms-normfrom
vllm-ir-oot
Open

vllm ir oot kernel registration#4
wxsIcey wants to merge 3 commits intoluka/vllm-ir/rms-normfrom
vllm-ir-oot

Conversation

@wxsIcey
Copy link
Owner

@wxsIcey wxsIcey commented Mar 9, 2026

Purpose

Currently, vllm/kernels/__init__.py unconditionally imported all kernel modules (vllm_c, aiter_ops, oink_ops), making it impossible for OOT platforms to control which IR implementations get registered without modifying vLLM internals.

This change introduces Platform.import_kernels() as the single entry point for kernel module loading. Each platform subclass overrides this method to import only the kernel registration modules relevant to its hardware, while OOT platforms can do the same without touching any vLLM-internal files.

wxsIcey added 3 commits March 9, 2026 06:09
Signed-off-by: Icey <1790571317@qq.com>
Signed-off-by: Icey <1790571317@qq.com>
Signed-off-by: Icey <1790571317@qq.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant