aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph2/GraphContext.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2018-03-21 17:52:35 +0000
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:49:16 +0000
commit9a8c672dd6eb21448fbfb4b636104323d128de88 (patch)
tree345615b62b3941f2c2db442758fcc12269137156 /arm_compute/graph2/GraphContext.h
parent36a559e49a3d5b832b1cffd47f2298f452616bb9 (diff)
downloadComputeLibrary-9a8c672dd6eb21448fbfb4b636104323d128de88.tar.gz
COMPMID-1011: Create struct for the graph config parameters
Change-Id: I9c164a817c0cc5f264a5c71a59256dacc6314cb0 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/125456 Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/graph2/GraphContext.h')
-rw-r--r--arm_compute/graph2/GraphContext.h25
1 files changed, 8 insertions, 17 deletions
diff --git a/arm_compute/graph2/GraphContext.h b/arm_compute/graph2/GraphContext.h
index 72ed96e7a0..f38e25dd61 100644
--- a/arm_compute/graph2/GraphContext.h
+++ b/arm_compute/graph2/GraphContext.h
@@ -56,26 +56,18 @@ public:
GraphContext &operator=(const GraphContext &) = delete;
/** Default move assignment operator */
GraphContext &operator=(GraphContext &&) = default;
- /** Enables tuning
+ /** Graph configuration accessor
*
- * @param[in] enable_tuning Enables tuning if true
- */
- void enable_tuning(bool enable_tuning);
- /** Checks if tuning is enabled
- *
- * @return True if tuning is enabled else false
- */
- bool is_tuning_enabled() const;
- /** Enables memory management
+ * @note Every alteration has to be done before graph finalization
*
- * @param[in] enable_mm Enables mm if true
+ * @return The graph configuration
*/
- void enable_memory_managenent(bool enable_mm);
- /** Checks if memory management is enabled
+ const GraphConfig &config() const;
+ /** Sets graph configuration
*
- * @return True if memory management is enabled else false
+ * @param[in] config Configuration to use
*/
- bool is_memory_management_enabled();
+ void set_config(const GraphConfig &config);
/** Inserts a memory manager context
*
* @param[in] memory_ctx Memory manage context
@@ -94,8 +86,7 @@ public:
void finalize();
private:
- bool _tunable; /**< Specifies if the Graph should use a tunable object */
- bool _memory_managed; /**< Specifies if the Graph should use a memory managed */
+ GraphConfig _config; /**< Graph configuration */
std::map<Target, MemoryManagerContext> _memory_managers; /**< Memory managers for each target */
};
} // namespace graph2