CMakeLists.txt 1.3 KB
Newer Older
Y
Yu Yang 已提交
1
cc_library(var_handle SRCS var_handle.cc DEPS place)
Y
Yu Yang 已提交
2
cc_library(op_handle_base SRCS op_handle_base.cc DEPS var_handle device_context)
Y
Yu Yang 已提交
3
cc_library(scale_loss_grad_op_handle SRCS scale_loss_grad_op_handle.cc DEPS op_handle_base scope lod_tensor ddim memory)
Y
Yu Yang 已提交
4
cc_library(fetch_op_handle SRCS fetch_op_handle.cc DEPS op_handle_base scope lod_tensor ddim memory)
Y
Yu Yang 已提交
5 6
nv_library(nccl_all_reduce_op_handle SRCS nccl_all_reduce_op_handle.cc DEPS op_handle_base scope lod_tensor ddim memory
        dynload_cuda)
Y
Yu Yang 已提交
7
cc_library(computation_op_handle SRCS computation_op_handle.cc DEPS framework_proto scope place operator op_registry)
Y
Yu Yang 已提交
8 9

cc_library(ssa_graph SRCS ssa_graph.cc DEPS var_handle op_handle_base)
Y
Yu Yang 已提交
10
cc_library(ssa_graph_builder SRCS ssa_graph_builder.cc DEPS ssa_graph)
Y
Yu Yang 已提交
11 12 13 14 15 16

if(WITH_GPU)
    set(multi_devices_graph_builder_deps nccl_all_reduce_op_handle)
else()
    set(multi_devices_graph_builder_deps)
endif()
Y
Yu Yang 已提交
17
cc_library(multi_devices_graph_builder SRCS multi_devices_graph_builder.cc DEPS ssa_graph_builder computation_op_handle
Y
Yu Yang 已提交
18
            scale_loss_grad_op_handle ${multi_devices_graph_builder_deps})
Y
Yu Yang 已提交
19 20 21
cc_library(ssa_graph_executor SRCS ssa_graph_executor.cc DEPS ssa_graph)
cc_library(threaded_ssa_graph_executor SRCS threaded_ssa_graph_executor.cc DEPS fetch_op_handle ssa_graph_executor scope
        simple_threadpool device_context)