blob: 654355b7f513f6999acadf64b1159cadf03077f5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# Unit tests
set(UNIT_TESTS
graph_tests
operator_tests
rnn_tests
attention_tests
fastopt_tests
utils_tests
)
foreach(test ${UNIT_TESTS})
add_executable("run_${test}" run_tests.cpp "${test}.cpp")
if(CUDA_FOUND)
target_link_libraries("run_${test}" ${EXT_LIBS} marian ${EXT_LIBS} marian_cuda ${EXT_LIBS} Catch)
else(CUDA_FOUND)
target_link_libraries("run_${test}" marian ${EXT_LIBS} Catch)
endif(CUDA_FOUND)
add_test(NAME ${test} COMMAND "run_${test}")
endforeach(test)
|