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 @@ -37,6 +37,7 @@ limitations under the License. */
37
37
#include " paddle/fluid/memory/allocation/allocator_strategy.h"
38
38
#include " paddle/fluid/operators/activation_op.h"
39
39
#include " paddle/fluid/operators/reader/lod_tensor_blocking_queue.h"
40
+ #include " paddle/fluid/platform/cpu_info.h"
40
41
#include " paddle/fluid/platform/enforce.h"
41
42
#include " paddle/fluid/platform/init.h"
42
43
#include " paddle/fluid/platform/place.h"
@@ -86,6 +87,9 @@ bool IsCompiledWithDIST() {
86
87
}
87
88
88
89
PYBIND11_PLUGIN (core) {
90
+ // Not used, just make sure cpu_info.cc is linked.
91
+ paddle::platform::CpuTotalPhysicalMemory ();
92
+
89
93
paddle::memory::allocation::UseAllocatorStrategyGFlag ();
90
94
py::module m (" core" , " C++ core of PaddlePaddle" );
91
95
Original file line number Diff line number Diff line change @@ -23,7 +23,9 @@ if(NOT WITH_DISTRIBUTE)
23
23
LIST (REMOVE_ITEM TEST_OPS test_dist_text_classification )
24
24
endif (NOT WITH_DISTRIBUTE )
25
25
26
- if (${CUDNN_MAJOR_VERSION} VERSION_LESS 7 )
26
+ if (NOT ${WITH_GPU} )
27
+ LIST (REMOVE_ITEM TEST_OPS test_conv2d_fusion_op )
28
+ elseif (${CUDNN_MAJOR_VERSION} VERSION_LESS 7 )
27
29
LIST (REMOVE_ITEM TEST_OPS test_conv2d_fusion_op )
28
30
endif ()
29
31
You can’t perform that action at this time.
0 commit comments