CMakeLists.txt 3.1 KB
Newer Older
Z
Zeng Jinle 已提交
1
cc_library(imperative_flag SRCS flags.cc DEPS gflags flags)
J
Jiabin Yang 已提交
2
cc_library(var_helper SRCS var_helper.cc DEPS tensor pten_api)
Q
QingshuChen 已提交
3
IF(WITH_XPU)
J
Jiabin Yang 已提交
4
cc_library(prepared_operator SRCS prepared_operator.cc DEPS xpu_op_list proto_desc operator device_context lod_tensor selected_rows_utils var_type_traits op_kernel_type data_transform nan_inf_utils pten_api pten pten_utils var_helper)
Q
QingshuChen 已提交
5
ELSE()
J
Jiabin Yang 已提交
6
cc_library(prepared_operator SRCS prepared_operator.cc DEPS proto_desc operator device_context lod_tensor selected_rows_utils var_type_traits op_kernel_type data_transform nan_inf_utils pten_api pten pten_utils var_helper)
Q
QingshuChen 已提交
7
ENDIF()
J
Jiabin Yang 已提交
8
cc_library(layer SRCS layer.cc DEPS prepared_operator math_function imperative_flag variable_helper op_registry var_helper pten_api)
9
add_subdirectory(jit)
J
Jiabin Yang 已提交
10 11
cc_library(amp SRCS amp_auto_cast.cc DEPS layer var_helper)
cc_library(tracer SRCS tracer.cc DEPS layer engine program_desc_tracer amp denormal garbage_collector var_helper)
12 13
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)
Z
Zeng Jinle 已提交
14
cc_library(imperative_profiler SRCS profiler.cc DEPS flags)
15
if(NOT WIN32)
16
    if(WITH_NCCL OR WITH_RCCL)
17
        cc_library(imperative_all_reduce SRCS all_reduce.cc DEPS collective_helper device_context selected_rows_utils tensor)
18
        cc_library(nccl_context SRCS nccl_context.cc DEPS collective_helper device_context imperative_all_reduce var_type_traits)
19 20 21 22 23 24
        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()
25
    endif()
26 27 28 29
    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()
30 31 32 33
    if(WITH_ASCEND_CL)
        cc_library(hccl_context SRCS hccl_context.cc DEPS collective_helper device_context tensor var_type_traits)
        cc_library(reducer SRCS reducer.cc DEPS layer)
    endif()
K
kuizhiqing 已提交
34 35 36
    if(WITH_NCCL OR WITH_RCCL OR WITH_XPU_BKCL OR WITH_ASCEND_CL)
        cc_library(heter_ccl_context SRCS heter_ccl_context.cc DEPS collective_helper device_context tensor var_type_traits)
    endif()
37
    cc_library(data_loader SRCS data_loader.cc DEPS enforce)
38
endif(NOT WIN32)
39 40
if(WITH_GLOO)
    cc_library(imperative_gloo_context SRCS gloo_context.cc DEPS collective_helper device_context tensor var_type_traits)
41
    if ( WIN32 OR (NOT (WITH_NCCL OR WITH_RCCL OR WITH_XPU_BKCL OR WITH_ASCEND_CL) ))
42 43 44
        cc_library(reducer SRCS reducer.cc DEPS layer)
    endif()
endif()
J
Jiabin Yang 已提交
45

46
if(NOT WITH_ASCEND_CL)
47
cc_library(gradient_accumulator SRCS gradient_accumulator.cc DEPS blas operator lod_tensor selected_rows_utils selected_rows_functor var_type_traits layer math_function)
48
else()
49
cc_library(gradient_accumulator SRCS gradient_accumulator.cc DEPS blas operator lod_tensor selected_rows_utils selected_rows_functor var_type_traits layer math_function npu_op_runner)
50
endif()
51

J
Jiabin Yang 已提交
52
add_subdirectory(tests)