|
@@ -103,6 +103,8 @@ Status KnownNodeTask::Init(TaskContext &context) { |
|
|
void *buffer = nullptr; |
|
|
void *buffer = nullptr; |
|
|
size_t total_mem_size = davinci_model_->TotalMemSize(); |
|
|
size_t total_mem_size = davinci_model_->TotalMemSize(); |
|
|
size_t total_zero_copy_size = davinci_model_->TotalZeroCopySize(); |
|
|
size_t total_zero_copy_size = davinci_model_->TotalZeroCopySize(); |
|
|
|
|
|
GELOGI("####KnownNodeTask::Init total mem size is %lu, total zero size is %lu.", |
|
|
|
|
|
total_mem_size, total_zero_copy_size); |
|
|
if (total_mem_size != 0 && total_mem_size > total_zero_copy_size) { |
|
|
if (total_mem_size != 0 && total_mem_size > total_zero_copy_size) { |
|
|
RECORD_EXECUTION_EVENT(context.GetExecutionContext(), context.GetNodeName(), |
|
|
RECORD_EXECUTION_EVENT(context.GetExecutionContext(), context.GetNodeName(), |
|
|
"[KnownNodeTask_AllocateWorkspace] Start"); |
|
|
"[KnownNodeTask_AllocateWorkspace] Start"); |
|
|