Skip to content

Commit 7d4b9fa

Browse files
author
Varun Sundar Rabindranath
committed
fix comment
Signed-off-by: Varun Sundar Rabindranath <[email protected]>
1 parent 80b647c commit 7d4b9fa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/lora/ops/triton_ops/kernel_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def mm_k(a_ptr, b_ptr, ak_stride, bk_stride, offset_k, K: tl.constexpr,
2727
BLOCK_M: M dimension of the output block m x n
2828
BLOCK_N: N dimension of the output block m x n
2929
BLOCK_K: K dimension atom
30-
EVEN_K: True iff the blocks of A and B can be loaded without any
30+
EVEN_K: True if the blocks of A and B can be loaded without any
3131
masking.
3232
SPLIT_K: Parameter signifying parallelism in the K dimension.
3333
CAST_TYPE: if True, cast the values from the A matrix to the B
@@ -127,7 +127,7 @@ def do_expand_kernel(
127127
offset_n = tl.arange(0, BLOCK_N) + pid_n * BLOCK_N
128128
rbn = tl.max_contiguous(tl.multiple_of(offset_n % N, BLOCK_N), BLOCK_N)
129129

130-
# Identify the row pointers of A and column pointers of B
130+
# 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 +
133133
offset_k[None, :] * input_d2_stride, )
@@ -213,7 +213,7 @@ def do_shrink_kernel(
213213
offset_n = tl.arange(0, BLOCK_N) + pid_n * BLOCK_N
214214
rbn = tl.max_contiguous(tl.multiple_of(offset_n % N, BLOCK_N), BLOCK_N)
215215

216-
# Identify the row pointers of A and column pointers of B
216+
# Identify A and B block pointers
217217
offset_k = pid_sk * BLOCK_K + tl.arange(0, BLOCK_K)
218218
a_ptr = (input_ptr + ram[:, None] * input_d0_stride +
219219
offset_k[None, :] * input_d1_stride)

0 commit comments

Comments
 (0)