From 3a8484c4eb1749382ea61677d761df0792fdcf0f Mon Sep 17 00:00:00 2001
From: cyber-pioneer <116002591+cyber-pioneer@users.noreply.github.com>
Date: Thu, 15 Jun 2023 11:26:34 +0800
Subject: [PATCH] fix batch_norm optest code (#54661)

---
 test/legacy_test/CMakeLists.txt             | 1 +
 test/legacy_test/test_batch_norm_op_prim.py | 4 +---
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt
index 6ce669cff60..281ed6382b5 100644
--- a/test/legacy_test/CMakeLists.txt
+++ b/test/legacy_test/CMakeLists.txt
@@ -1185,6 +1185,7 @@ set(TEST_CINN_OPS
     test_clip_op
     test_scatter_op
     test_gather_op
+    test_batch_norm_op_prim
     test_layer_norm_op
     test_cast_op
     test_dropout_op
diff --git a/test/legacy_test/test_batch_norm_op_prim.py b/test/legacy_test/test_batch_norm_op_prim.py
index 17148541bd4..b0c93cfb4a8 100644
--- a/test/legacy_test/test_batch_norm_op_prim.py
+++ b/test/legacy_test/test_batch_norm_op_prim.py
@@ -1,4 +1,4 @@
-#   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");
 # you may not use this file except in compliance with the License.
@@ -130,8 +130,6 @@ class TestBatchNormOp(OpTest):
                 check_prim=True,
                 only_check_prim=True,
             )
-        # restore init config
-        self.initConfig()
 
     def initConfig(self):
         self.rev_comp_atol = 1e-5
-- 
GitLab