Skip to content

Add ability to replace oot ops when using lora#37181

Merged
yaochengji merged 1 commit intovllm-project:mainfrom
kyuyeunk:oot_lora_replace
Mar 17, 2026
Merged

Add ability to replace oot ops when using lora#37181
yaochengji merged 1 commit intovllm-project:mainfrom
kyuyeunk:oot_lora_replace

Conversation

@kyuyeunk
Copy link
Copy Markdown
Contributor

@kyuyeunk kyuyeunk commented Mar 16, 2026

Purpose

Add ability to support lora when using custom ops. Otherwise, custom ops will not be replaced with Lora layers.

Test Plan

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.
  • (Optional) Release notes update. If your change is user facing, please update the release notes draft in the Google Doc.

@kyuyeunk kyuyeunk requested a review from jeejeelee as a code owner March 16, 2026 11:15
@kyuyeunk kyuyeunk changed the title Replace oot layers for lora Add ability to replace oot ops when using lora Mar 16, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a mechanism to support LoRA with custom out-of-tree (OOT) operators by using a get_new_class_type helper for type checks. The approach is sound, but the implementation is incomplete as several type checks in vllm/lora/layers/column_parallel_linear.py were not updated. This will prevent LoRA from working with custom OOT layers for those specific layer types. I've added a detailed comment pointing out the missed locations.

@kyuyeunk kyuyeunk force-pushed the oot_lora_replace branch 2 times, most recently from 8247ff1 to fe5134d Compare March 16, 2026 11:25
Copy link
Copy Markdown
Collaborator

@yaochengji yaochengji left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

others, LGTM!

@yaochengji yaochengji enabled auto-merge (squash) March 16, 2026 21:29
@github-actions github-actions bot added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 16, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
auto-merge was automatically disabled March 16, 2026 22:23

Head branch was pushed to by a user without write access

@kyuyeunk
Copy link
Copy Markdown
Contributor Author

CI has passed. @yaochengji, auto-merge was disabled after i did rebase, can you trigger merge for this pr again? thanks!

@yaochengji yaochengji merged commit 0a0a1a1 into vllm-project:main Mar 17, 2026
61 checks passed
Lucaskabela pushed a commit to Lucaskabela/vllm that referenced this pull request Mar 17, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
andylolu2 pushed a commit to andylolu2/vllm that referenced this pull request Mar 18, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
wendyliu235 pushed a commit to wendyliu235/vllm-public that referenced this pull request Mar 18, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
fxdawnn pushed a commit to fxdawnn/vllm that referenced this pull request Mar 19, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
khairulkabir1661 pushed a commit to khairulkabir1661/vllm that referenced this pull request Mar 27, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
Monishver11 pushed a commit to Monishver11/vllm that referenced this pull request Mar 27, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
Signed-off-by: Monishver Chandrasekaran <monishverchandrasekaran@gmail.com>
JiantaoXu pushed a commit to JiantaoXu/vllm that referenced this pull request Mar 28, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
vrdn-23 pushed a commit to vrdn-23/vllm that referenced this pull request Mar 30, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
Signed-off-by: Vinay Damodaran <vrdn@hey.com>
EricccYang pushed a commit to EricccYang/vllm that referenced this pull request Apr 1, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
Signed-off-by: EricccYang <yangyang4991@gmail.com>
liuchenbing2026 pushed a commit to liuchenbing2026/vllm that referenced this pull request Apr 4, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
big-yellow-duck pushed a commit to EmbeddedLLM/vllm that referenced this pull request Apr 8, 2026
Signed-off-by: Kyuyeun Kim <kyuyeunk@google.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants