diff options
Diffstat (limited to 'src/gpu/cl/ClContext.h')
-rw-r--r-- | src/gpu/cl/ClContext.h | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/src/gpu/cl/ClContext.h b/src/gpu/cl/ClContext.h index dd6699a0c9..2a0d4ee1c8 100644 --- a/src/gpu/cl/ClContext.h +++ b/src/gpu/cl/ClContext.h @@ -44,6 +44,7 @@ public: * @param[in] options Creational options */ explicit ClContext(const AclContextOptions *options); + /** Extract MLGO heuristics * * @return Heuristics tree @@ -55,6 +56,13 @@ public: * @return the cl context used */ ::cl::Context cl_ctx(); + + /** Underlying cl device accessor + * + * @return the cl device used + */ + ::cl::Device cl_dev(); + /** Update/inject an underlying cl context object * * @warning Context will be able to set if the object doesn't have any pending reference to other objects @@ -67,10 +75,12 @@ public: // Inherrited methods overridden ITensorV2 *create_tensor(const AclTensorDescriptor &desc, bool allocate) override; + IQueue *create_queue(const AclQueueOptions *options) override; private: mlgo::MLGOHeuristics _mlgo_heuristics; - ::cl::Context _cl_context; + ::cl::Context _cl_ctx; + ::cl::Device _cl_dev; }; } // namespace opencl } // namespace gpu |