Skip to content

GitLab

"vllm/vscode:/vscode.git/clone" did not exist on "a1239b53e809ef2deb3df6818d7dca00669eec0c"
Unverified Commit 90bd2ae1 authored by Varun Sundar Rabindranath's avatar Varun Sundar Rabindranath Committed by GitHub
Browse files

[Bugfix] LoRA - Retire unused maxnreg LoRA kernel argument (#17677)

parent 5941e0b7
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
vllm/lora/ops/triton_ops/lora_expand_op.py
View file @ 90bd2ae1
......@@ -204,7 +204,6 @@ def _lora_expand(
NUM_WARPS = 4
NUM_CTAS = 1
NUM_STAGES = 2
MAX_NREG = None
EVEN_K = K % BLOCK_K == 0 # type: ignore
......@@ -258,7 +257,6 @@ def _lora_expand(
num_warps=NUM_WARPS,
num_ctas=NUM_CTAS,
num_stages=NUM_STAGES,
maxnreg=MAX_NREG,
)
return
......
vllm/lora/ops/triton_ops/lora_shrink_op.py
View file @ 90bd2ae1
......@@ -168,7 +168,6 @@ def _lora_shrink(
NUM_WARPS = 4
NUM_CTAS = 1
NUM_STAGES = 2
MAX_NREG = None
EVEN_K = K % (BLOCK_K * SPLIT_K) == 0 # type: ignore
......@@ -213,7 +212,6 @@ def _lora_shrink(
num_warps=NUM_WARPS,
num_ctas=NUM_CTAS,
num_stages=NUM_STAGES,
maxnreg=MAX_NREG,
)
return
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment