Browse Source

!14352 fix log file write incomplete

From: @yuzhenhua666
Reviewed-by: @wuxuejian,@c_34
Signed-off-by: @wuxuejian
pull/14352/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
b2448481e8
2 changed files with 17 additions and 2 deletions
  1. +9
    -1
      model_zoo/official/cv/faster_rcnn/ascend310_infer/src/AclProcess.cpp
  2. +8
    -1
      model_zoo/official/cv/maskrcnn/ascend310_infer/src/AclProcess.cpp

+ 9
- 1
model_zoo/official/cv/faster_rcnn/ascend310_infer/src/AclProcess.cpp View File

@@ -169,7 +169,15 @@ int AclProcess::WriteResult(const std::string& imageFile) {
std::cout << "open result file " << outFileName << " failed" << std::endl;
return INVALID_POINTER;
}
fwrite(resHostBuf, output_size, sizeof(char), outputFile);
size_t size = fwrite(resHostBuf, sizeof(char), output_size, outputFile);
if (size != output_size) {
fclose(outputFile);
outputFile = nullptr;
std::cout << "write result file " << outFileName << " failed, write size[" << size <<
"] is smaller than output size[" << output_size << "], maybe the disk is full." << std::endl;
return ERROR;
}

fclose(outputFile);
outputFile = nullptr;
} catch (std::exception &e) {


+ 8
- 1
model_zoo/official/cv/maskrcnn/ascend310_infer/src/AclProcess.cpp View File

@@ -169,7 +169,14 @@ int AclProcess::WriteResult(const std::string& imageFile) {
std::cout << "open result file " << outFileName << " failed" << std::endl;
return INVALID_POINTER;
}
fwrite(resHostBuf, output_size, sizeof(char), outputFile);
size_t size = fwrite(resHostBuf, sizeof(char), output_size, outputFile);
if (size != output_size) {
fclose(outputFile);
outputFile = nullptr;
std::cout << "write result file " << outFileName << " failed, write size[" << size <<
"] is smaller than output size[" << output_size << "], maybe the disk is full." << std::endl;
return ERROR;
}
fclose(outputFile);
outputFile = nullptr;
} catch (std::exception &e) {


Loading…
Cancel
Save