CMakeLists.txt 462 字节
Newer Older
1
add_subdirectory(api)
2
add_subdirectory(accumulation)
3
add_subdirectory(tests)
4

5
cc_library(autograd_meta SRCS autograd_meta.cc DEPS pten pten_api)
6
cc_library(grad_node_info SRCS grad_node_info.cc DEPS pten pten_api)
7
cc_library(grad_tensor_holder SRCS grad_tensor_holder.cc DEPS grad_node_info gradient_accumulation)
8
cc_library(utils SRCS utils.cc DEPS pten pten_api global_utils layer proto_desc operator op_registry variable_helper memcpy scale_op autograd_meta)