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
@@ -1,12 +1,16 @@
 #include <iostream>

+#include "tkdnn.h"
 #include "Network.h"

 namespace tkDNN {

 Network::Network() {

-    std::cout<<"New NETWORK with CUDNN v"<<float(cudnnGetVersion())/1000<<"\n";
+    float tk_ver = float(tkDNN::getVersion())/1000;
+    float cu_ver = float(cudnnGetVersion())/1000;
+
+    std::cout<<"New NETWORK (tkDNN v"<<tk_ver<<", CUDNN v"<<cu_ver<<")\n";
    dataType = CUDNN_DATA_FLOAT;
    tensorFormat = CUDNN_TENSOR_NCHW;