Skip to content

Commit

Permalink
fix python lint
Browse files Browse the repository at this point in the history
  • Loading branch information
zhangzefeng92 committed Mar 27, 2024
1 parent fce081f commit 7e59912
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deeplink_ext/internlm_ops/rms_norm/deeplink.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def backward(ctx, grad_output):
grad_input = torch.empty_like(hidden_states)
grad_weight = torch.empty_like(weight)
grad_bias = torch.empty_like(bias)

ext.rms_norm_backward(
grad_input,
grad_weight,
Expand Down

0 comments on commit 7e59912

Please sign in to comment.