Skip to content

Commit e418862

Browse files
committed
fix a error in unit tests
1 parent 47c02b5 commit e418862

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/paddle/fluid/tests/unittests/test_bilinear_interp_op.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import unittest
1616
import numpy as np
1717
from op_test import OpTest
18+
import paddle.fluid.core as core
1819

1920

2021
def bilinear_interp_np(input, out_h, out_w, out_size):
@@ -138,7 +139,7 @@ def setUp(self):
138139
self.outputs = {'Out': output_np}
139140

140141
def test_check_output(self):
141-
self.check_output(atol=1)
142+
self.check_output_with_place(place=core.CPUPlace(), atol=1)
142143

143144
def init_test_case(self):
144145
self.input_shape = [1, 3, 9, 6]

0 commit comments

Comments
 (0)