diff options
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.hpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.hpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.hpp b/src/backends/backendsCommon/WorkloadData.hpp index c563626b28..952ddc323a 100644 --- a/src/backends/backendsCommon/WorkloadData.hpp +++ b/src/backends/backendsCommon/WorkloadData.hpp @@ -29,15 +29,23 @@ struct QueueDescriptor { std::vector<ITensorHandle*> m_Inputs; std::vector<ITensorHandle*> m_Outputs; + void* m_AdditionalInfoObject; void ValidateInputsOutputs(const std::string& descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const; + template<typename T> + const T* GetAdditionalInformation() + { + return static_cast<T*>(m_AdditionalInfoObject); + } protected: ~QueueDescriptor() = default; - QueueDescriptor() = default; + QueueDescriptor() + : m_AdditionalInfoObject(nullptr) + {} QueueDescriptor(QueueDescriptor const&) = default; QueueDescriptor& operator=(QueueDescriptor const&) = default; }; |