Skip to content

Commit 4198686

Browse files
authored
doc: fix the math display of group gemm operator (#292)
1 parent e252c94 commit 4198686

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/flashinfer/group_gemm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def forward(
8484
8585
.. math::
8686
87-
y[i] = x[i] \times W[weight_indices[i]]
87+
y[i] = x[i] \times W[\text{weight_indices}[i]]
8888
8989
We use Ragged Tensor to represent the input tensor :attr:`x` and the output tensor :attr:`y`, and each x[i]
9090
is a segment of the concatenated tensor. Please see :ref:`Ragged Tensor tutorial <ragged-layout>` for more details.

0 commit comments

Comments
 (0)