update unitest

PaddlePaddle · Nov 10, 2023 · 5485b41 · 5485b41
1 parent 39b3b9e
commit 5485b41
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/xpu/test_layer_norm_op_xpu.py b/test/xpu/test_layer_norm_op_xpu.py
@@ -44,6 +44,8 @@ def ref_layer_norm(x, scale, bias, epsilon, begin_norm_axis=1):
     if bias is not None:
         y = y + bias.reshape([1, right])
     x.shape, y.shape = x_shape, x_shape
+    mean.shape = x_shape[0:begin_norm_axis]
+    variance.shape = x_shape[0:begin_norm_axis]
     return y, mean, variance