Skip to content

Commit 53cb220

Browse files
authored
try to resolve the inference_pass_test, test=develop (#24192)
1 parent d3c9db7 commit 53cb220

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

python/paddle/fluid/tests/unittests/ir/inference/inference_pass_test.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -29,19 +29,19 @@
2929

3030

3131
class InferencePassTest(unittest.TestCase):
32-
@classmethod
33-
def setUpClass(cls):
34-
cls.main_program = fluid.Program()
35-
cls.startup_program = fluid.Program()
36-
cls.feeds = None
37-
cls.fetch_list = None
38-
39-
cls.enable_mkldnn = False
40-
cls.enable_trt = False
41-
cls.trt_parameters = None
42-
cls.enable_lite = False
43-
cls.lite_parameters = None
44-
cls.path = "./inference_pass/"
32+
def __init__(self, methodName='runTest'):
33+
super(InferencePassTest, self).__init__(methodName)
34+
self.main_program = fluid.Program()
35+
self.startup_program = fluid.Program()
36+
self.feeds = None
37+
self.fetch_list = None
38+
39+
self.enable_mkldnn = False
40+
self.enable_trt = False
41+
self.trt_parameters = None
42+
self.enable_lite = False
43+
self.lite_parameters = None
44+
self.path = "./inference_pass/" + self.__class__.__name__ + "/"
4545
np.random.seed(1)
4646
random.seed(1)
4747

0 commit comments

Comments
 (0)