aboutsummaryrefslogtreecommitdiff
path: root/src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp')
-rw-r--r--src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp b/src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp
index 549c6d4abb..f14f66d1bd 100644
--- a/src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp
+++ b/src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp
@@ -287,15 +287,20 @@ Status ClWorkloadRuntime::configure(const GpuWorkloadSketch &sketch)
switch(stage)
{
case UnitWorkloadStage::Stage::Run:
+ {
_impl->_kernels.emplace(work.id(), std::move(k));
break;
+ }
case UnitWorkloadStage::Stage::Prepare:
+ {
_impl->_kernels_prep.emplace(work.id(), std::move(k));
break;
+ }
default:
+ {
ARM_COMPUTE_ERROR("Invalid unit workload stage");
+ }
}
- break;
}
// Create auxiliary tensor objects
create_aux_tensors(&_impl->_aux_tensors, _impl->_source_code);