We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e252c94 commit 4198686Copy full SHA for 4198686
python/flashinfer/group_gemm.py
@@ -84,7 +84,7 @@ def forward(
84
85
.. math::
86
87
- y[i] = x[i] \times W[weight_indices[i]]
+ y[i] = x[i] \times W[\text{weight_indices}[i]]
88
89
We use Ragged Tensor to represent the input tensor :attr:`x` and the output tensor :attr:`y`, and each x[i]
90
is a segment of the concatenated tensor. Please see :ref:`Ragged Tensor tutorial <ragged-layout>` for more details.
0 commit comments