Skip to content

Commit 5d8e1c9

Browse files
oteroantoniogomServerAI
andauthored
[Bugfix] Support triton==3.3.0+git95326d9f for RTX 5090 (Unsloth + vLLM compatibility) (#15471)
Co-authored-by: ServerAI <ai@exc-mad-ai.com>
1 parent 0a049c7 commit 5d8e1c9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/lora/ops/triton_ops/kernel_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def do_expand_kernel(
130130
# Identify A and B block pointers
131131
offset_k = tl.arange(0, BLOCK_K)
132132
a_ptr = (cur_input_ptr + ram[:, None] * input_d1_stride +
133-
offset_k[None, :] * input_d2_stride, )
133+
offset_k[None, :] * input_d2_stride)
134134
b_ptr = (cur_lora_ptr + cur_lora_d0_stride * lora_index +
135135
offset_k[:, None] * cur_lora_d2_stride +
136136
rbn[None, :] * cur_lora_d1_stride)

0 commit comments

Comments
 (0)