!8377 fix bug in sub_graph_kernel

From: @hangangqiang
Reviewed-by: @zhanghaibo5
Signed-off-by:
pull/8377/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 6d4edc6dfb

@ -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) {

@ -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;
}
};

Loading…
Cancel
Save