diff options
Diffstat (limited to 'src/dynamic_fusion/runtime/gpu')
-rw-r--r-- | src/dynamic_fusion/runtime/gpu/cl/ClKernelRuntime.h | 3 | ||||
-rw-r--r-- | src/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.cpp | 7 |
2 files changed, 8 insertions, 2 deletions
diff --git a/src/dynamic_fusion/runtime/gpu/cl/ClKernelRuntime.h b/src/dynamic_fusion/runtime/gpu/cl/ClKernelRuntime.h index acc2380031..4787acabcd 100644 --- a/src/dynamic_fusion/runtime/gpu/cl/ClKernelRuntime.h +++ b/src/dynamic_fusion/runtime/gpu/cl/ClKernelRuntime.h @@ -35,7 +35,8 @@ namespace experimental { namespace dynamic_fusion { -struct GpuKernelSourceCode; +/** Forward declaration */ +class GpuKernelSourceCode; /** OpenCL runtime to run a single kernel */ class ClKernelRuntime final : public opencl::IClKernel 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); |