diff options
Diffstat (limited to 'src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp')
-rw-r--r-- | src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp | 7 |
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); |