diff --git a/tensorflow/lite/micro/tools/benchmarking/generic_model_benchmark.cc b/tensorflow/lite/micro/tools/benchmarking/generic_model_benchmark.cc index 4249deb320c..c1db8fa9cf3 100644 --- a/tensorflow/lite/micro/tools/benchmarking/generic_model_benchmark.cc +++ b/tensorflow/lite/micro/tools/benchmarking/generic_model_benchmark.cc @@ -79,7 +79,8 @@ void SetRandomInput(const uint32_t random_seed, bool ReadFile(const char* file_name, void* buffer, size_t buffer_size) { std::unique_ptr file(fopen(file_name, "rb"), fclose); - fread(buffer, sizeof(char), buffer_size, file.get()); + const size_t bytes_read = + fread(buffer, sizeof(char), buffer_size, file.get()); if (ferror(file.get())) { MicroPrintf("Unable to read model file: %d\n", ferror(file.get())); return false; @@ -88,6 +89,10 @@ bool ReadFile(const char* file_name, void* buffer, size_t buffer_size) { MicroPrintf("Model buffer is too small for the model.\n"); return false; } + if (bytes_read == 0) { + MicroPrintf("No bytes read from model file.\n"); + return false; + } return true; }