diff --git a/mindspore/lite/src/sub_graph_kernel.cc b/mindspore/lite/src/sub_graph_kernel.cc
index 3c01f7fccf..119b8422c5 100644
--- a/mindspore/lite/src/sub_graph_kernel.cc
+++ b/mindspore/lite/src/sub_graph_kernel.cc
@@ -160,7 +160,9 @@ int CpuSubGraph::Prepare() {
 
 void CpuFp16SubGraph::FreeOriginInputData() {
   for (auto *data_store : this->origin_input_data_) {
-    MS_ASSERT(data_store != nullptr);
+    if (data_store == nullptr) {
+      continue;
+    }
     // free data in data_store
     if (data_store->data_ != nullptr) {
       if (data_store->allocator_ == nullptr) {
diff --git a/mindspore/lite/src/sub_graph_kernel.h b/mindspore/lite/src/sub_graph_kernel.h
index 0418170805..e8de8e9b23 100644
--- a/mindspore/lite/src/sub_graph_kernel.h
+++ b/mindspore/lite/src/sub_graph_kernel.h
@@ -65,19 +65,19 @@ struct DataStore {
   lite::Allocator *allocator_ = nullptr;
   static DataStore *CreateDataStore(void *data = nullptr, lite::Allocator *data_allocator = nullptr,
                                     lite::Allocator *allocator = nullptr) {
-    DataStore *tensor_data = nullptr;
+    DataStore *data_store = nullptr;
     if (allocator == nullptr) {
-      tensor_data = static_cast<DataStore *>(malloc(sizeof(DataStore)));
+      data_store = static_cast<DataStore *>(malloc(sizeof(DataStore)));
     } else {
-      tensor_data = static_cast<DataStore *>(allocator->Malloc(sizeof(DataStore)));
+      data_store = static_cast<DataStore *>(allocator->Malloc(sizeof(DataStore)));
     }
-    if (tensor_data == nullptr) {
-      MS_LOG(ERROR) << "Malloc tensor_data failed";
+    if (data_store == nullptr) {
+      MS_LOG(ERROR) << "Malloc data_store failed";
       return nullptr;
     }
-    tensor_data->data_ = data;
-    tensor_data->allocator_ = data_allocator;
-    return tensor_data;
+    data_store->data_ = data;
+    data_store->allocator_ = data_allocator;
+    return data_store;
   }
 };