CMakeLists.txt 753 字节
Newer Older
1 2 3
if(WIN32)
    cc_test(nccl_context_test SRCS nccl_context_test.cc DEPS device_context)
else()
4 5 6
    if (WITH_NCCL)
        cc_test(nccl_context_test SRCS nccl_context_test.cc DEPS nccl_context)
    endif()
7 8 9
endif(WIN32)


10
cc_test(test_gradient_accmulator SRCS test_gradient_accmulator.cc DEPS memcpy selected_rows selected_rows_functor gradient_accumulator)
11
cc_test(test_layer SRCS test_layer.cc DEPS layer proto_desc operator op_registry variable_helper mul_op memcpy)
12
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 已提交
13
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)