diff --git a/include/tkdnn.h b/include/tkdnn.h index 372ca48..243ace7 100644 --- a/include/tkdnn.h +++ b/include/tkdnn.h @@ -5,13 +5,4 @@ #include "Layer.h" #include "NetworkRT.h" -namespace tkDNN { - - /** - Return the tkDNN version - */ - int getVersion() { - - return 100; - } -} \ No newline at end of file +#define TKDNN_VERSION 200 diff --git a/src/Network.cpp b/src/Network.cpp index fa97c25..7157293 100644 --- a/src/Network.cpp +++ b/src/Network.cpp @@ -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"<