schihei/ascend-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update msame print TXT

Browse Source
This commit is contained in:
chen68
2020-10-19 01:15:20 +08:00
parent 9af6ae2b35
commit ad220e1a7e
1 changed files with 64 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msame/src/model_process.cpp
+52 -24
View File
@@ -440,13 +440,19 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
} }
if (f_isTXT) { if (f_isTXT) {
ofstream outstr(times + "/" + modelName + "_output_" + to_string(i) + ".txt", ios::out); ofstream outstr(times + "/" + modelName + "_output_" + to_string(i) + ".txt", ios::out);
int amount_onebatch = 1;
for (int j = 1; j < dim->dimCount; j++) {
amount_onebatch *= dim->dims[j];
}
switch (datatype) { switch (datatype) {
case 0: case 0:
for (int i = 0; i < len / sizeof(float); i++) { for (int i = 0; i < len / sizeof(float); i++) {
float out = *((float*)outData + i); float out = *((float*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -456,8 +462,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(aclFloat16); i++) { for (int i = 0; i < len / sizeof(aclFloat16); i++) {
aclFloat16 out = *((aclFloat16*)outData + i); aclFloat16 out = *((aclFloat16*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -467,8 +475,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(int8_t); i++) { for (int i = 0; i < len / sizeof(int8_t); i++) {
int8_t out = *((int8_t*)outData + i); int8_t out = *((int8_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -478,8 +488,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(int); i++) { for (int i = 0; i < len / sizeof(int); i++) {
int out = *((int*)outData + i); int out = *((int*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -489,8 +501,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(uint8_t); i++) { for (int i = 0; i < len / sizeof(uint8_t); i++) {
uint8_t out = *((uint8_t*)outData + i); uint8_t out = *((uint8_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -500,8 +514,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(int16_t); i++) { for (int i = 0; i < len / sizeof(int16_t); i++) {
int16_t out = *((int16_t*)outData + i); int16_t out = *((int16_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -511,8 +527,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(uint16_t); i++) { for (int i = 0; i < len / sizeof(uint16_t); i++) {
uint16_t out = *((uint16_t*)outData + i); uint16_t out = *((uint16_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -522,8 +540,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(uint32_t); i++) { for (int i = 0; i < len / sizeof(uint32_t); i++) {
uint32_t out = *((uint32_t*)outData + i); uint32_t out = *((uint32_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -533,8 +553,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(int64_t); i++) { for (int i = 0; i < len / sizeof(int64_t); i++) {
int64_t out = *((int64_t*)outData + i); int64_t out = *((int64_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -544,8 +566,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(uint64_t); i++) { for (int i = 0; i < len / sizeof(uint64_t); i++) {
uint64_t out = *((uint64_t*)outData + i); uint64_t out = *((uint64_t*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -555,8 +579,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(double); i++) { for (int i = 0; i < len / sizeof(double); i++) {
double out = *((double*)outData + i); double out = *((double*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
@@ -566,8 +592,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
for (int i = 0; i < len / sizeof(bool); i++) { for (int i = 0; i < len / sizeof(bool); i++) {
int out = *((bool*)outData + i); int out = *((bool*)outData + i);
outstr << out << " "; outstr << out << " ";
for (int j = 0; j < dim->dimCount; j++) { if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
if (i != 0 && i % dim->dims[j] == 0 && dim->dims[j] > 10) { outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }