Add typo checker in pre-commit (#6179)

Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
2025-05-11 12:55:00 +08:00
parent de167cf5fa
commit 2ce8793519
99 changed files with 154 additions and 144 deletions
@@ -428,7 +428,7 @@ class Llama4DecoderLayer(nn.Module):
        # Fully Connected
        hidden_states = self.feed_forward(hidden_states, forward_batch)

-        # TODO(ch-wan): ues reduce-scatter in MLP to avoid this scatter
+        # TODO(ch-wan): use reduce-scatter in MLP to avoid this scatter
        # Scatter
        if self.dp_size != 1:
            # important: forward batch.gathered_buffer is used both after scatter and after gather.