|
@@ -286,9 +286,9 @@ Status SubgraphExecutor::PrepareForExecution(GraphExecutionContext *ctx, NodeSta |
|
|
auto shared_task_context = std::shared_ptr<TaskContext>(unique_task_context.release()); |
|
|
auto shared_task_context = std::shared_ptr<TaskContext>(unique_task_context.release()); |
|
|
node_state.SetTaskContext(shared_task_context); |
|
|
node_state.SetTaskContext(shared_task_context); |
|
|
GE_CHK_RT_RET(rtCtxSetCurrent(ctx->rt_context)); |
|
|
GE_CHK_RT_RET(rtCtxSetCurrent(ctx->rt_context)); |
|
|
RECORD_COMPILE_EVENT(ctx, node_item.NodeItem().c_str(), "[UpdateTilingData] start"); |
|
|
|
|
|
|
|
|
RECORD_COMPILE_EVENT(ctx, node_item.NodeName().c_str(), "[UpdateTilingData] start"); |
|
|
GE_CHK_STATUS_RET_NOLOG(task->UpdateTilingData(*shared_task_context)); // update op_desc before alloc ws |
|
|
GE_CHK_STATUS_RET_NOLOG(task->UpdateTilingData(*shared_task_context)); // update op_desc before alloc ws |
|
|
RECORD_COMPILE_EVENT(ctx, node_item.NodeItem().c_str(), "[UpdateTilingData] end"); |
|
|
|
|
|
|
|
|
RECORD_COMPILE_EVENT(ctx, node_item.NodeName().c_str(), "[UpdateTilingData] end"); |
|
|
return SUCCESS; |
|
|
return SUCCESS; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|