From 7b607dc5058d2f6a398cfac4d2eab65d415c3733 Mon Sep 17 00:00:00 2001 From: Anthony Barbier Date: Fri, 13 Jul 2018 15:55:24 +0100 Subject: COMPMID-1400: Add command line option to specify the tuner's config file Change-Id: Ib597e0dff4c8c01f7e6bd46d03824beef4bc1e9a Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/139923 Tested-by: Jenkins Reviewed-by: Georgios Pinitas --- arm_compute/graph/Types.h | 9 +++++---- arm_compute/graph/backends/CL/CLDeviceBackend.h | 1 + 2 files changed, 6 insertions(+), 4 deletions(-) (limited to 'arm_compute/graph') diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h index 9cfede9cf3..c5b7fb1c51 100644 --- a/arm_compute/graph/Types.h +++ b/arm_compute/graph/Types.h @@ -77,10 +77,11 @@ class TensorDescriptor; /** Graph configuration structure */ struct GraphConfig { - bool use_function_memory_manager{ true }; /**< Use a memory manager to manage per-funcion auxilary memory */ - bool use_transition_memory_manager{ true }; /**< Use a memory manager to manager transition buffer memory */ - bool use_tuner{ false }; /**< Use a tuner in tunable backends */ - int num_threads{ -1 }; /**< Number of threads to use (thread capable backends), if 0 the backend will auto-initialize, if -1 the backend will stay as it is. */ + bool use_function_memory_manager{ true }; /**< Use a memory manager to manage per-funcion auxilary memory */ + bool use_transition_memory_manager{ true }; /**< Use a memory manager to manager transition buffer memory */ + bool use_tuner{ false }; /**< Use a tuner in tunable backends */ + int num_threads{ -1 }; /**< Number of threads to use (thread capable backends), if 0 the backend will auto-initialize, if -1 the backend will stay as it is. */ + std::string tuner_file{ "acl_tuner.csv" }; /**< File to load/store tuning values from */ }; /**< Device target types */ diff --git a/arm_compute/graph/backends/CL/CLDeviceBackend.h b/arm_compute/graph/backends/CL/CLDeviceBackend.h index 0b45dfe479..c1a6a28e6c 100644 --- a/arm_compute/graph/backends/CL/CLDeviceBackend.h +++ b/arm_compute/graph/backends/CL/CLDeviceBackend.h @@ -66,6 +66,7 @@ private: bool _initialized; /**< Flag that specifies if the backend has been default initialized */ CLTuner _tuner; /**< CL kernel tuner */ std::unique_ptr _allocator; /**< CL buffer affinity allocator */ + std::string _tuner_file; /** Filename to load/store the tuner's values from */ }; } // namespace backends } // namespace graph -- cgit v1.2.1