diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/GpuKernelComponentStream.h')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/GpuKernelComponentStream.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/GpuKernelComponentStream.h b/src/dynamic_fusion/sketch/gpu/GpuKernelComponentStream.h index cbaa7c297b..ba2503a938 100644 --- a/src/dynamic_fusion/sketch/gpu/GpuKernelComponentStream.h +++ b/src/dynamic_fusion/sketch/gpu/GpuKernelComponentStream.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022 Arm Limited. + * Copyright (c) 2022-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -49,10 +49,11 @@ class GpuKernelComponentStream public: /** Constructor * + * @param[in] context @ref GpuWorkloadContext to be used throughout the stream * @param[in] services @ref GpuComponentServices to be used throughout the stream * @param[in] mem_map @ref MemoryDescriptor map used to assemble the @ref GpuWorkloadSourceCode */ - GpuKernelComponentStream(GpuComponentServices *services, const MemoryDescriptorMap &mem_map); + GpuKernelComponentStream(GpuWorkloadContext *context, GpuComponentServices *services, const MemoryDescriptorMap &mem_map); /** Allow instances of this class to be copy constructed */ GpuKernelComponentStream(const GpuKernelComponentStream &stream) = default; /** Allow instances of this class to be copied */ @@ -78,6 +79,7 @@ public: bool add_component(IGpuKernelComponent *component); private: + GpuWorkloadContext *_context; GpuComponentServices *_services; std::vector<GpuKernelComponentGroup> _component_groups{}; MemoryDescriptorMap _mem_map{}; |