From 0c6ce880580a92744e5a55d63ef050ca7b342300 Mon Sep 17 00:00:00 2001
From: shenwei41 <shenwei41@huawei.com>
Date: Tue, 8 Sep 2020 11:00:10 +0800
Subject: [PATCH] Alarm modification

---
 .../minddata/dataset/engine/datasetops/source/mindrecord_op.cc  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.cc b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.cc
index 8bc314b0a..d64c28f72 100644
--- a/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.cc
+++ b/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.cc
@@ -52,6 +52,7 @@ MindRecordOp::Builder::Builder() : build_dataset_file_({}) {
   build_rows_per_buffer_ = cfg->rows_per_buffer();
   build_op_connector_queue_size_ = cfg->op_connector_size();
   builder_num_workers_ = 0;
+  build_load_dataset_ = false;
   build_num_padded_ = 0;
   build_sample_ = nullptr;
 }
@@ -344,6 +345,7 @@ Status MindRecordOp::LoadTensorRow(TensorRow *tensor_row, const std::vector<uint
     DataType type = column.type();
 
     // Set shape
+    CHECK_FAIL_RETURN_UNEXPECTED(column_data_type_size != 0, "The divisor cannot be 0.");
     auto num_elements = n_bytes / column_data_type_size;
     if (type == DataType::DE_STRING) {
       std::string s{data, data + n_bytes};
-- 
GitLab