tkDNN library version 0.1

2017-06-29 12:50:10 +00:00
parent 8bbfeeb19b
commit d80ec7f6ee
3 changed files with 7 additions and 4 deletions
@@ -14,5 +14,4 @@ add_library(tkDNN SHARED src/Layer.cpp src/LayerWgs.cpp
 target_link_libraries(tkDNN kernels ${CUDA_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_TOOLKIT_ROOT_DIR}/lib/libcudnn.so)

 add_executable(tkDNNtest tests/test.cpp)
-message(${CUDA_LIBRARIES})
 target_link_libraries(tkDNNtest tkDNN)