|
|
|
@ -120,11 +120,11 @@ Status ModelHelper::SaveModelTbeKernel(std::shared_ptr<OmFileSaveHelper> &om_fil
|
|
|
|
|
TBEKernelStore tbe_kernel_store = ge_model->GetTBEKernelStore();
|
|
|
|
|
GELOGD("TBE_KERNELS size is %zu", tbe_kernel_store.DataSize());
|
|
|
|
|
if (tbe_kernel_store.DataSize() > 0) {
|
|
|
|
|
GE_CHK_STATUS_RET(SaveModelPartition(om_file_save_helper,
|
|
|
|
|
ModelPartitionType::TBE_KERNELS,
|
|
|
|
|
ge_model->GetTBEKernelStore().Data(),
|
|
|
|
|
ge_model->GetTBEKernelStore().DataSize(), model_index),
|
|
|
|
|
"Add tbe kernel partition failed");
|
|
|
|
|
GE_CHK_STATUS_RET(
|
|
|
|
|
SaveModelPartition(om_file_save_helper, ModelPartitionType::TBE_KERNELS,
|
|
|
|
|
ge_model->GetTBEKernelStore().Data(),
|
|
|
|
|
ge_model->GetTBEKernelStore().DataSize(), model_index),
|
|
|
|
|
"Add tbe kernel partition failed");
|
|
|
|
|
}
|
|
|
|
|
// no need to check value, DATA->NetOutput
|
|
|
|
|
(void)tbe_kernel_store.Load(tbe_kernel_store.Data(), tbe_kernel_store.DataSize());
|
|
|
|
|