From 7e59912ba48046b8bfdd2a6f4fa4ed825858592a Mon Sep 17 00:00:00 2001 From: zhangzefeng92 Date: Wed, 27 Mar 2024 10:20:39 +0800 Subject: [PATCH] fix python lint --- deeplink_ext/internlm_ops/rms_norm/deeplink.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeplink_ext/internlm_ops/rms_norm/deeplink.py b/deeplink_ext/internlm_ops/rms_norm/deeplink.py index 82142e17..793bc02e 100644 --- a/deeplink_ext/internlm_ops/rms_norm/deeplink.py +++ b/deeplink_ext/internlm_ops/rms_norm/deeplink.py @@ -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,