File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
paddle/fluid/operators/reduce_ops Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change 1
1
include (operators )
2
- register_operators ()
2
+ if (WITH_GPU )
3
+ register_operators (DEPS cub )
4
+ else ()
5
+ register_operators ()
6
+ endif ()
3
7
4
8
if (WITH_GPU )
5
9
file (GLOB OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR} " "*.part.cu" )
Original file line number Diff line number Diff line change @@ -54,7 +54,7 @@ ELSE(WIN32)
54
54
DEPENDS copy_paddle_pybind ${FLUID_CORE} framework_py_proto profiler_py_proto ${PY_FILES} ${external_project_dependencies} ${COPY_PADDLE_MASTER} )
55
55
ENDIF ()
56
56
57
- set (paddle_python_deps ${PADDLE_PYTHON_BUILD_DIR} /.timestamp ${MKL_DEPENDS} )
57
+ set (paddle_python_deps ${PADDLE_PYTHON_BUILD_DIR} /.timestamp ${MKL_DEPENDS} ${external_project_dependencies} )
58
58
add_custom_target (paddle_python ALL DEPENDS ${paddle_python_deps} )
59
59
60
60
set (PADDLE_PYTHON_PACKAGE_DIR ${CMAKE_CURRENT_BINARY_DIR} /dist/ )
You can’t perform that action at this time.
0 commit comments