From e4188621ca253cfc5edeb63158f98acb0ca0432a Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Wed, 20 Jun 2018 15:59:55 +0800 Subject: [PATCH] fix a error in unit tests --- python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py b/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py index 2935c71dd3..b04f25ef87 100644 --- a/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py +++ b/python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py @@ -15,6 +15,7 @@ import unittest import numpy as np from op_test import OpTest +import paddle.fluid.core as core def bilinear_interp_np(input, out_h, out_w, out_size): @@ -138,7 +139,7 @@ class TestBilinearInterpOpUint8(OpTest): self.outputs = {'Out': output_np} def test_check_output(self): - self.check_output(atol=1) + self.check_output_with_place(place=core.CPUPlace(), atol=1) def init_test_case(self): self.input_shape = [1, 3, 9, 6] -- GitLab