CMakeLists.txt 1.3 KB
Newer Older
1
set(eager_deps phi_api phi_dygraph_api hook_utils tensor_utils utils global_utils backward phi_tensor tracer layer autograd_meta grad_node_info grad_tensor_holder accumulation_node custom_operator_node)
2

3
set(fluid_deps tracer layer proto_desc operator op_registry variable_helper memcpy)
4
set(generated_deps final_dygraph_function final_dygraph_node dygraph_function dygraph_node)
5

6
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
7 8 9 10
    message("Performing Eager Dygraph Auto Code Generation")
    add_subdirectory(auto_code_generator)
endif()

11
add_subdirectory(api)
12
add_subdirectory(accumulation)
13
add_subdirectory(custom_operator)
W
wanghuancoder 已提交
14 15
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
    add_subdirectory(pylayer)
16 17 18
    cc_library(grad_tensor_holder SRCS grad_tensor_holder.cc DEPS grad_node_info gradient_accumulator)
    add_dependencies(grad_tensor_holder eager_final_state_codegen)
    cc_library(backward SRCS backward.cc DEPS grad_tensor_holder utils autograd_meta grad_node_info)
W
wanghuancoder 已提交
19
endif()
20

21
cc_library(grad_node_info SRCS grad_node_info.cc DEPS phi_api phi_tensor)
22

23 24
cc_library(autograd_meta SRCS autograd_meta.cc DEPS phi_api phi_tensor)
cc_library(utils SRCS utils.cc DEPS phi_api phi_tensor global_utils layer proto_desc operator op_registry variable_helper memcpy scale_op autograd_meta hook_utils)
25

26 27 28
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
    add_subdirectory(tests)
endif()