CMakeLists.txt

set(eager_deps
    phi_api
    phi_dygraph_api
    hook_utils
    tensor_utils
    utils
    global_utils
    backward
    phi_tensor
    tracer
    layer
    autograd_meta
    eager_nan_inf_utils
    grad_node_info
    grad_tensor_holder
    accumulation_node
    custom_operator_node)

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

if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
  message("Performing Eager Dygraph Auto Code Generation")
  add_subdirectory(auto_code_generator)
endif()

add_subdirectory(api)
add_subdirectory(accumulation)
add_subdirectory(custom_operator)
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
  add_subdirectory(pylayer)
  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 switch_autotune)
endif()

cc_library(
  eager_nan_inf_utils
  SRCS nan_inf_utils.cc
  DEPS phi_tensor nan_inf_utils enforce)
cc_library(
  grad_node_info
  SRCS grad_node_info.cc
  DEPS phi_api phi_tensor)

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)

if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
  add_subdirectory(tests)
endif()