Skip to content

Commit 2996c9a

Browse files
committed
Fix indentation in rms_norm.py - replace tab with spaces
1 parent 62d0c3e commit 2996c9a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/liger_kernel/ops/rms_norm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ def _block_rms_norm_backward_kernel(
351351

352352
# calculate the gradient of W
353353
if casting_mode == _CASTING_MODE_LLAMA:
354-
# TODO(tcc): use tl.sum(..., dtype=tl.float32) once we upgrade to triton>=3.3.0
354+
# TODO(tcc): use tl.sum(..., dtype=tl.float32) once we upgrade to triton>=3.3.0
355355
dW_row += tl.sum((dY_row * (X_row * rstd_row[:, None]).to(X_dtype)).to(tl.float32), 0)
356356
else:
357357
# here X_row is already in fp32 (see previous if block)

0 commit comments

Comments
 (0)