File tree Expand file tree Collapse file tree 2 files changed +7
-1
lines changed
python/paddle/fluid/tests/unittests Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -28,6 +28,10 @@ list(REMOVE_ITEM TEST_OPS test_cond_op) # FIXME(qijun): https://github.com/Paddl
28
28
29
29
list (REMOVE_ITEM TEST_OPS op_test ) # op_test is a helper python file, not a test
30
30
list (REMOVE_ITEM TEST_OPS decorators ) # decorators is a helper python file, not a test
31
+ if (APPLE )
32
+ # this op is not support on mac
33
+ list (REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op )
34
+ endif ()
31
35
32
36
function (py_test_modules TARGET_NAME )
33
37
if (WITH_TESTING )
Original file line number Diff line number Diff line change 13
13
# limitations under the License.
14
14
15
15
from __future__ import print_function
16
-
16
+ import os
17
17
import paddle .fluid as fluid
18
18
import paddle
19
19
import numpy as np
@@ -41,6 +41,8 @@ def fake_data_generator():
41
41
self .data_file_name , reader , feeder )
42
42
43
43
def setUp (self ):
44
+ # set parallel threads to fit 20 batches in line 49
45
+ os .environ ['CPU_NUM' ] = str (20 )
44
46
self .use_cuda = fluid .core .is_compiled_with_cuda ()
45
47
self .data_file_name = './reader_reset_test.recordio'
46
48
self .ins_shape = [3 ]
You can’t perform that action at this time.
0 commit comments