Skip to content

Commit 7fe3c2d

Browse files
committed
lint fix
1 parent e09c1b7 commit 7fe3c2d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/gdn/example_chunk_o_bwd.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,9 @@ def kernel(
254254
# for i_kv in T.Parallel(block_DK * block_DV):
255255
# dg_last_fragment[i_kv] = h_shared[i_kv // block_DV, i_kv % block_DV] * dh_shared[i_kv // block_DV, i_kv % block_DV]
256256
for i_kv in T.Parallel(block_DK * block_DV):
257-
dg_last_fragment[i_kv] = h_shared[i_kv // block_DV, i_kv % block_DV] * dh_shared[i_kv // block_DV, i_kv % block_DV]
257+
dg_last_fragment[i_kv] = h_shared[i_kv // block_DV, i_kv %
258+
block_DV] * dh_shared[i_kv // block_DV,
259+
i_kv % block_DV]
258260
T.reduce_sum(dg_last_fragment, dg_last_fragment_scalar, dim=-1, clear=False)
259261
dg_last_local[0] += dg_last_fragment_scalar[0]
260262

0 commit comments

Comments
 (0)