File tree Expand file tree Collapse file tree 2 files changed +14
-0
lines changed
python/paddle/fluid/tests/unittests Expand file tree Collapse file tree 2 files changed +14
-0
lines changed Original file line number Diff line number Diff line change @@ -245,9 +245,17 @@ op_library(channel_send_op DEPS concurrency)
245
245
op_library (channel_recv_op DEPS concurrency )
246
246
247
247
list (REMOVE_ITEM GENERAL_OPS ${DEPS_OPS} )
248
+
249
+ # The fully connected layer is deleted when the WITH_MKLDNN flag is OFF
250
+ # Because the fully connected layer has only one MKLDNN's operator
251
+ if (NOT WITH_MKLDNN )
252
+ list (REMOVE_ITEM GENERAL_OPS fc_op )
253
+ endif (NOT WITH_MKLDNN )
254
+
248
255
foreach (src ${GENERAL_OPS} )
249
256
op_library (${src} )
250
257
endforeach ()
258
+
251
259
file (APPEND ${pybind_file} "USE_OP(less_than);\n USE_OP(logical_and);\n USE_NO_KERNEL_OP(read_from_array);\n " )
252
260
253
261
add_subdirectory (reader )
Original file line number Diff line number Diff line change 1
1
file (GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR} " "test_*.py" )
2
2
string (REPLACE ".py" "" TEST_OPS "${TEST_OPS} " )
3
3
4
+ # The fully connected test is removed whe the WITH_MKLDNN flag is OFF
5
+ # Because the fully connected layer has only one kernel (MKLDNN)
6
+ if (NOT WITH_MKLDNN )
7
+ list (REMOVE_ITEM TEST_OPS test_fc_op )
8
+ endif (NOT WITH_MKLDNN )
9
+
4
10
if (NOT WITH_DISTRIBUTE )
5
11
list (REMOVE_ITEM TEST_OPS test_recv_op )
6
12
endif (NOT WITH_DISTRIBUTE )
You can’t perform that action at this time.
0 commit comments