version fix

2017-08-22 01:43:36 -07:00
parent 5595b8037b
commit c41a0a09a6
2 changed files with 2 additions and 11 deletions
@@ -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