diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-05-02 14:07:55 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:51:17 +0000 |
commit | e043767d068da389308507011d944e6db9e4d676 (patch) | |
tree | 30c8965d8d03d141c7630420c6e945f78485efc7 /arm_compute/runtime/CL/functions/CLGEMM.h | |
parent | 019634f8befde24b19bae9b749e75a9f3ae44801 (diff) | |
download | ComputeLibrary-e043767d068da389308507011d944e6db9e4d676.tar.gz |
COMPMID-920: Introduce prepare() stage
Change-Id: I08ddb7f6e061178e7566518b48e4e18f8f078596
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/129825
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLGEMM.h')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLGEMM.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/runtime/CL/functions/CLGEMM.h b/arm_compute/runtime/CL/functions/CLGEMM.h index c5d7b86384..60ff32c6fa 100644 --- a/arm_compute/runtime/CL/functions/CLGEMM.h +++ b/arm_compute/runtime/CL/functions/CLGEMM.h @@ -100,6 +100,7 @@ public: // Inherited methods overridden: void run() override; + void prepare() override; private: CLMemoryGroup _memory_group; @@ -112,8 +113,8 @@ private: const ICLTensor *_original_b; bool _is_interleaved_transposed; bool _run_addition; - bool _is_first_run; bool _reshape_b_only_on_first_run; + bool _is_prepared; }; } |