fix some typos (#24616)
Signed-off-by: co63oc <co63oc@users.noreply.github.com>
This commit is contained in:
@@ -179,7 +179,7 @@ def chunk_local_cumsum_vector(
|
||||
def grid(meta):
|
||||
return (triton.cdiv(meta['S'], meta['BS']), NT, B * H)
|
||||
|
||||
# keep cummulative normalizer in fp32
|
||||
# keep cumulative normalizer in fp32
|
||||
# this kernel is equivalent to
|
||||
# g = g.view(B, H, NT, BT, -1).cumsum(-2).view(B, H, T, -1)
|
||||
chunk_local_cumsum_vector_kernel[grid](g_org,
|
||||
|
||||
Reference in New Issue
Block a user