version fix
This commit is contained in:
+1
-10
@@ -5,13 +5,4 @@
|
|||||||
#include "Layer.h"
|
#include "Layer.h"
|
||||||
#include "NetworkRT.h"
|
#include "NetworkRT.h"
|
||||||
|
|
||||||
namespace tkDNN {
|
#define TKDNN_VERSION 200
|
||||||
|
|
||||||
/**
|
|
||||||
Return the tkDNN version
|
|
||||||
*/
|
|
||||||
int getVersion() {
|
|
||||||
|
|
||||||
return 100;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
+1
-1
@@ -10,7 +10,7 @@ namespace tkDNN {
|
|||||||
Network::Network(dataDim_t input_dim) {
|
Network::Network(dataDim_t input_dim) {
|
||||||
this->input_dim = 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;
|
float cu_ver = float(cudnnGetVersion())/1000;
|
||||||
|
|
||||||
std::cout<<"New NETWORK (tkDNN v"<<tk_ver
|
std::cout<<"New NETWORK (tkDNN v"<<tk_ver
|
||||||
|
|||||||
Reference in New Issue
Block a user