未验证 提交 3a8484c4 编写于 作者: C cyber-pioneer 提交者: GitHub

fix batch_norm optest code (#54661)

上级 7fef4ee9
...@@ -1185,6 +1185,7 @@ set(TEST_CINN_OPS ...@@ -1185,6 +1185,7 @@ set(TEST_CINN_OPS
test_clip_op test_clip_op
test_scatter_op test_scatter_op
test_gather_op test_gather_op
test_batch_norm_op_prim
test_layer_norm_op test_layer_norm_op
test_cast_op test_cast_op
test_dropout_op test_dropout_op
......
# Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved. # Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
# #
# Licensed under the Apache License, Version 2.0 (the "License"); # Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License. # you may not use this file except in compliance with the License.
...@@ -130,8 +130,6 @@ class TestBatchNormOp(OpTest): ...@@ -130,8 +130,6 @@ class TestBatchNormOp(OpTest):
check_prim=True, check_prim=True,
only_check_prim=True, only_check_prim=True,
) )
# restore init config
self.initConfig()
def initConfig(self): def initConfig(self):
self.rev_comp_atol = 1e-5 self.rev_comp_atol = 1e-5
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册