mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

dla

Browse Source
This commit is contained in:
Roberto Cavicchioli
2019-03-06 16:39:36 +01:00
parent c22219ad16
commit 3714155809
3 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Network.cpp
+9 -1
View File
@@ -24,12 +24,20 @@ Network::Network(dataDim_t input_dim) {
num_layers = 0;
fp16 = false;
if(const char* env_p = std::getenv("TKDNN_MODE"))
dla = false;
if(const char* env_p = std::getenv("TKDNN_MODE")) {
if(strcmp(env_p, "FP16") == 0)
fp16 = true;
else if(strcmp(env_p, "DLA") == 0) {
dla = true;
fp16 = true;
}
}
if(fp16)
std::cout<<COL_REDB<<"!! FP16 INERENCE ENABLED !!"<<COL_END<<"\n";
if(dla)
std::cout<<COL_GREENB<<"!! DLA INERENCE ENABLED !!"<<COL_END<<"\n";
}
Network::~Network() {