|
@@ -50,8 +50,8 @@ Status HybridModelExecutor::Execute(HybridModelExecutor::ExecuteArgs &args) { |
|
|
auto root_graph_item = model_->GetRootGraphItem(); |
|
|
auto root_graph_item = model_->GetRootGraphItem(); |
|
|
GE_CHECK_NOTNULL(root_graph_item); |
|
|
GE_CHECK_NOTNULL(root_graph_item); |
|
|
|
|
|
|
|
|
GE_CHK_RT_RET(rtMemcpy(context_.global_step, sizeof(uint64_t), &context_.iteration, |
|
|
|
|
|
sizeof(uint64_t), RT_MEMCPY_HOST_TO_DEVICE)); |
|
|
|
|
|
|
|
|
GE_CHK_RT_RET(rtMemcpyAsync(context_.global_step, sizeof(uint64_t), &context_.iteration, |
|
|
|
|
|
sizeof(uint64_t), RT_MEMCPY_HOST_TO_DEVICE_EX, context_.stream)); |
|
|
SubgraphExecutor executor(model_->GetRootGraphItem(), &context_); |
|
|
SubgraphExecutor executor(model_->GetRootGraphItem(), &context_); |
|
|
auto ret = ExecuteGraphInternal(executor, args); |
|
|
auto ret = ExecuteGraphInternal(executor, args); |
|
|
Cleanup(); |
|
|
Cleanup(); |
|
|