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

Fix getNetworkRTName (works also on Xavier)

Browse Source 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2020-04-02 10:15:30 +02:00
parent 79a05a8159
commit e4900120db
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Network.cpp
+4 -4
View File
@@ -124,24 +124,24 @@ const char *Network::getNetworkRTName(char *network_name){
int network_name_len = strlen(network_name);
char *RTName = (char *)malloc((network_name_len + 9)*sizeof(char));
if (fp16){
strcat(RTName, network_name);
strcpy(RTName, network_name);
strcat(RTName, "_fp16.rt");
RTName[network_name_len + 7] = '\0';
}
else if (dla){
strcat(RTName, network_name);
strcpy(RTName, network_name);
strcat(RTName, "_dla.rt");
RTName[network_name_len + 6] = '\0';
}
else if (int8){
strcat(RTName, network_name);
strcpy(RTName, network_name);
strcat(RTName, "_int8.rt");
RTName[network_name_len + 8] = '\0';
}
else{
strcat(RTName, network_name);
strcpy(RTName, network_name);
strcat(RTName, "_fp32.rt");
RTName[network_name_len + 8] = '\0';
}