提交 59bbf075 编写于 作者: D dingminghui 提交者: jackzhang235

fix: fix extra bridges not exist in subgraph_mlu

上级 1bf17174
...@@ -102,6 +102,10 @@ if(WITH_GRPC) ...@@ -102,6 +102,10 @@ if(WITH_GRPC)
add_definitions(-DPADDLE_WITH_GRPC) add_definitions(-DPADDLE_WITH_GRPC)
endif(WITH_GRPC) endif(WITH_GRPC)
if(LITE_BUILD_EXTRA)
add_definitions(-DLITE_BUILD_EXTRA)
endif()
if(WITH_BRPC_RDMA) if(WITH_BRPC_RDMA)
add_definitions(-DPADDLE_WITH_BRPC_RDMA) add_definitions(-DPADDLE_WITH_BRPC_RDMA)
endif(WITH_BRPC_RDMA) endif(WITH_BRPC_RDMA)
......
...@@ -4,9 +4,6 @@ lite_cc_library(subgraph_detector ...@@ -4,9 +4,6 @@ lite_cc_library(subgraph_detector
lite_cc_library(subgraph_pass lite_cc_library(subgraph_pass
SRCS subgraph_pass.cc SRCS subgraph_pass.cc
DEPS mir_pass types context ${mir_fusers} subgraph_detector) DEPS mir_pass types context ${mir_fusers} subgraph_detector)
if (LITE_BUILD_EXTRA)
target_compile_definitions(subgraph_pass PUBLIC "-DLITE_BUILD_EXTRA")
endif()
if (WITH_TESTING AND NOT LITE_WITH_CUDA) if (WITH_TESTING AND NOT LITE_WITH_CUDA)
lite_cc_test(test_subgraph_detector lite_cc_test(test_subgraph_detector
SRCS subgraph_detector_test.cc SRCS subgraph_detector_test.cc
......
...@@ -57,11 +57,14 @@ set(mlu_subgraph_bridges ...@@ -57,11 +57,14 @@ set(mlu_subgraph_bridges
if (LITE_BUILD_EXTRA) if (LITE_BUILD_EXTRA)
lite_cc_library(subgraph_bridge_lrn_op_mlu SRCS lrn_op.cc DEPS ${subgraph_bridge_deps_mlu}) lite_cc_library(subgraph_bridge_lrn_op_mlu SRCS lrn_op.cc DEPS ${subgraph_bridge_deps_mlu})
list(APPEND mlu_subgraph_bridges subgraph_bridge_lrn_op_mlu)
lite_cc_library(subgraph_bridge_gather_op_mlu SRCS gather_op.cc DEPS ${subgraph_bridge_deps_mlu}) lite_cc_library(subgraph_bridge_gather_op_mlu SRCS gather_op.cc DEPS ${subgraph_bridge_deps_mlu})
list(APPEND mlu_subgraph_bridges subgraph_bridge_gather_op_mlu)
lite_cc_library(subgraph_bridge_norm_op_mlu SRCS norm_op.cc DEPS ${subgraph_bridge_deps_mlu}) lite_cc_library(subgraph_bridge_norm_op_mlu SRCS norm_op.cc DEPS ${subgraph_bridge_deps_mlu})
list(APPEND mlu_subgraph_bridges subgraph_bridge_norm_op_mlu) set(mlu_subgraph_bridges
"${mlu_subgraph_bridges}"
subgraph_bridge_lrn_op_mlu
subgraph_bridge_gather_op_mlu
subgraph_bridge_norm_op_mlu
CACHE INTERNAL "mlu_subgraph_bridges")
endif() endif()
lite_cc_library(subgraph_test_helper_mlu SRCS test_helper.cc DEPS ${mlu_subgraph_bridges}) lite_cc_library(subgraph_test_helper_mlu SRCS test_helper.cc DEPS ${mlu_subgraph_bridges})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册