CMakeLists.txt 1.4 KB
Newer Older
1 2 3
if(WIN32)
    cc_test(nccl_context_test SRCS nccl_context_test.cc DEPS device_context)
else()
4
    if (WITH_NCCL OR WITH_RCCL)
5
        cc_test(nccl_context_test SRCS nccl_context_test.cc DEPS nccl_context)
K
kuizhiqing 已提交
6 7
        cc_test(heter_ccl_context_test SRCS heter_ccl_context_test.cc DEPS heter_ccl_context nccl_context imperative_gloo_context gloo_context gloo_wrapper gloo fs shell)
        #set_tests_properties(heter_ccl_context_test PROPERTIES LABELS "RUN_TYPE=DIST")
8
    endif()
9 10 11
    if (WITH_XPU_BKCL)
        cc_test(bkcl_context_test SRCS bkcl_context_test.cc DEPS bkcl_context)
    endif()
12 13 14
endif(WIN32)


15
cc_test(test_gradient_accmulator SRCS test_gradient_accmulator.cc DEPS memcpy selected_rows selected_rows_functor gradient_accumulator math_function)
16
cc_test(test_layer SRCS test_layer.cc DEPS layer proto_desc operator op_registry variable_helper mul_op memcpy)
17
cc_test(test_prepare_op SRCS test_prepare_op.cc DEPS prepared_operator op_info split_op layer concat_and_split activation_op place)
H
hong 已提交
18
cc_test(test_tracer SRCS test_tracer.cc DEPS tracer layer proto_desc operator op_registry variable_helper mul_op reduce_sum_op elementwise_add_op memcpy)
19
cc_test(test_hooks SRCS test_hooks.cc DEPS tracer basic_engine layer proto_desc operator op_registry variable_helper mul_op elementwise_add_op memcpy)
20

21
if (WITH_NCCL OR WITH_RCCL OR WITH_XPU_BKCL)
22 23
cc_test(test_group SRCS test_group.cc DEPS reducer concat_and_split memcpy)
endif()