version fix

2017-08-22 01:43:36 -07:00
parent 5595b8037b
commit c41a0a09a6
2 changed files with 2 additions and 11 deletions
@@ -5,13 +5,4 @@
 #include "Layer.h"
 #include "NetworkRT.h"

-namespace tkDNN {
-
-    /**
-        Return the tkDNN version
-    */
-    int getVersion() {
-
-        return 100;
-    }
-}
+#define TKDNN_VERSION 200
@@ -10,7 +10,7 @@ namespace tkDNN {
 Network::Network(dataDim_t input_dim) {
    this->input_dim = input_dim;

-    float tk_ver = float(tkDNN::getVersion())/1000;
+    float tk_ver = float(TKDNN_VERSION)/1000;
    float cu_ver = float(cudnnGetVersion())/1000;

    std::cout<<"New NETWORK (tkDNN v"<<tk_ver