mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

better tests

Browse Source
This commit is contained in:
Francesco Gatti
2017-07-26 12:23:49 +02:00
parent e7a6f1fb6c
commit 483ffefc35
9 changed files with 170 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+4 -4
View File
@@ -13,8 +13,8 @@ add_library(tkDNN SHARED src/Layer.cpp src/LayerWgs.cpp
src/Network.cpp src/utils.cpp)
target_link_libraries(tkDNN kernels ${CUDA_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES} -lcudnn)
add_executable(tkDNNtest tests/test.cpp)
target_link_libraries(tkDNNtest tkDNN)
add_executable(test_simple tests/test/test.cpp)
target_link_libraries(test_simple tkDNN)
add_executable(mnist tests/mnist/test.cpp)
target_link_libraries(mnist tkDNN)
add_executable(test_mnist tests/mnist/test.cpp)
target_link_libraries(test_mnist tkDNN)