Skip to content

GitLab

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( ...@@ -204,7 +204,6 @@ def _lora_expand(
NUM_WARPS = 4 NUM_WARPS = 4
NUM_CTAS = 1 NUM_CTAS = 1
NUM_STAGES = 2 NUM_STAGES = 2
MAX_NREG = None
EVEN_K = K % BLOCK_K == 0 # type: ignore EVEN_K = K % BLOCK_K == 0 # type: ignore
...@@ -258,7 +257,6 @@ def _lora_expand( ...@@ -258,7 +257,6 @@ def _lora_expand(
num_warps=NUM_WARPS, num_warps=NUM_WARPS,
num_ctas=NUM_CTAS, num_ctas=NUM_CTAS,
num_stages=NUM_STAGES, num_stages=NUM_STAGES,
maxnreg=MAX_NREG,
) )
return return
......
vllm/lora/ops/triton_ops/lora_shrink_op.py
View file @ 90bd2ae1
...@@ -168,7 +168,6 @@ def _lora_shrink( ...@@ -168,7 +168,6 @@ def _lora_shrink(
NUM_WARPS = 4 NUM_WARPS = 4
NUM_CTAS = 1 NUM_CTAS = 1
NUM_STAGES = 2 NUM_STAGES = 2
MAX_NREG = None
EVEN_K = K % (BLOCK_K * SPLIT_K) == 0 # type: ignore EVEN_K = K % (BLOCK_K * SPLIT_K) == 0 # type: ignore
...@@ -213,7 +212,6 @@ def _lora_shrink( ...@@ -213,7 +212,6 @@ def _lora_shrink(
num_warps=NUM_WARPS, num_warps=NUM_WARPS,
num_ctas=NUM_CTAS, num_ctas=NUM_CTAS,
num_stages=NUM_STAGES, num_stages=NUM_STAGES,
maxnreg=MAX_NREG,
) )
return 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