CMakeLists.txt 1.0 KB
Newer Older
1
set(eager_deps phi_api hook_utils tensor_utils utils global_utils backward phi_tensor tracer layer autograd_meta grad_node_info grad_tensor_holder accumulation_node)
2 3 4
set(fluid_deps tracer layer proto_desc operator op_registry variable_helper memcpy)
set(generated_deps dygraph_function dygraph_node)

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

10
add_subdirectory(api)
11
add_subdirectory(accumulation)
12

13
cc_library(grad_node_info SRCS grad_node_info.cc DEPS phi_api phi_tensor)
14
cc_library(grad_tensor_holder SRCS grad_tensor_holder.cc DEPS grad_node_info gradient_accumulator)
15

16 17
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)
18
cc_library(backward SRCS backward.cc DEPS grad_tensor_holder utils autograd_meta grad_node_info)
19 20

add_subdirectory(tests)