未验证 提交 496de7f3 编写于 作者: Y yangjianfengo1 提交者: GitHub

Increase absolute error of test_group_norm_op (#55992)

* inplace tol

* code style
上级 ab8c3179
......@@ -357,37 +357,15 @@ class TestGroupNormFP16Op_With_NHWC(TestGroupNormFP16OP):
self.attrs['epsilon'] = 0.5
self.shape = (1, 100, 4, 4)
self.dtype = np.float16
input = np.sin(
np.arange(
self.shape[0] * self.shape[1] * self.shape[2] * self.shape[3]
)
)
input = np.transpose(input.reshape(self.shape), (0, 2, 3, 1)).astype(
self.dtype
)
scale = np.sin(np.arange(self.shape[1])).astype(self.dtype)
bias = np.sin(np.arange(self.shape[1])).astype(self.dtype)
output, mean, var = group_norm_naive(
input,
scale,
bias,
self.attrs['epsilon'],
self.attrs['groups'],
self.data_format,
)
self.inputs = {
'X': OpTest.np_dtype_to_fluid_dtype(input),
'Scale': OpTest.np_dtype_to_fluid_dtype(scale),
'Bias': OpTest.np_dtype_to_fluid_dtype(bias),
}
self.outputs = {'Y': output, 'Mean': mean, 'Variance': var}
self.attrs['data_layout'] = self.data_format
def test_check_output(self):
rtol = 2e-3
atol = 2e-3
inplace_atol = 2e-3
place = core.CUDAPlace(0)
self.check_output_with_place(place, rtol=rtol)
self.check_output_with_place(
place, rtol=rtol, atol=atol, inplace_atol=inplace_atol
)
class TestGroupNormBF16Op_With_NHWC(TestGroupNormBF16Op):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册