CMakeLists.txt 1.8 KB
Newer Older
1
cc_library(imperative_flag SRCS flags.cc DEPS gflags)
Z
Zeng Jinle 已提交
2

3
cc_library(prepared_operator SRCS prepared_operator.cc DEPS proto_desc operator device_context lod_tensor selected_rows var_type_traits op_kernel_type data_transform)
J
Jiabin Yang 已提交
4
cc_library(layer SRCS layer.cc DEPS prepared_operator math_function imperative_flag variable_helper op_registry)
5
add_subdirectory(jit)
6
cc_library(amp SRCS amp_auto_cast.cc DEPS layer )
7
cc_library(tracer SRCS tracer.cc DEPS layer engine program_desc_tracer amp denormal)
8 9
cc_library(basic_engine SRCS basic_engine.cc DEPS layer gradient_accumulator)
cc_library(engine SRCS basic_engine.cc partial_grad_engine.cc DEPS layer gradient_accumulator)
M
minqiyang 已提交
10
cc_library(imperative_profiler SRCS profiler.cc)
11
if(NOT WIN32)
12
    if(WITH_NCCL OR WITH_RCCL)
13
        cc_library(imperative_all_reduce SRCS all_reduce.cc DEPS collective_helper device_context selected_rows tensor)
14
        cc_library(nccl_context SRCS nccl_context.cc DEPS collective_helper device_context imperative_all_reduce var_type_traits)
15 16 17 18 19 20
        if(WITH_NCCL)
            nv_library(reducer SRCS reducer.cc reducer.cu DEPS layer imperative_all_reduce)
        endif()
        if(WITH_RCCL)
            hip_library(reducer SRCS reducer.cc reducer.cu DEPS layer imperative_all_reduce)
        endif()
21
    endif()
22 23 24 25
    if(WITH_XPU_BKCL)
        cc_library(bkcl_context SRCS bkcl_context.cc DEPS collective_helper device_context tensor var_type_traits)
        cc_library(reducer SRCS reducer.cc DEPS layer)
    endif()
26
    cc_library(data_loader SRCS data_loader.cc DEPS enforce)
27
endif(NOT WIN32)
J
Jiabin Yang 已提交
28

29 30
cc_library(gradient_accumulator SRCS gradient_accumulator.cc DEPS blas operator lod_tensor selected_rows selected_rows_functor var_type_traits layer math_function)

P
phlrain 已提交
31
#cc_binary(tracer_test SRCS tracer_test.cc DEPS tracer layer op_registry python pybind ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} profiler )
P
phlrain 已提交
32

J
Jiabin Yang 已提交
33
add_subdirectory(tests)