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 @@ -46,6 +46,7 @@ limitations under the License. */
46
46
#include " paddle/fluid/memory/allocation/allocator_strategy.h"
47
47
#include " paddle/fluid/operators/activation_op.h"
48
48
#include " paddle/fluid/operators/reader/lod_tensor_blocking_queue.h"
49
+ #include " paddle/fluid/platform/cpu_info.h"
49
50
#include " paddle/fluid/platform/enforce.h"
50
51
#include " paddle/fluid/platform/init.h"
51
52
#include " paddle/fluid/platform/place.h"
@@ -95,6 +96,9 @@ bool IsCompiledWithDIST() {
95
96
}
96
97
97
98
PYBIND11_PLUGIN (core) {
99
+ // Not used, just make sure cpu_info.cc is linked.
100
+ paddle::platform::CpuTotalPhysicalMemory ();
101
+
98
102
paddle::memory::allocation::UseAllocatorStrategyGFlag ();
99
103
py::module m (" core" , " C++ core of PaddlePaddle" );
100
104
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