diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2021-07-16 10:23:31 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-27 17:29:31 +0000 |
commit | d87aded57efb2997d486ffae9102eb79def60c99 (patch) | |
tree | 960eda814ef8002cf880d1e0798583590471d6b8 /src/runtime/gpu/cl/operators/ClGemm.h | |
parent | 4718706b1141d5cccb006a7f86d65c1fde6c54ff (diff) | |
download | ComputeLibrary-d87aded57efb2997d486ffae9102eb79def60c99.tar.gz |
Port CLGEMMConvolutionLayer
Details:
port CLWeightsReshapeKernel to ClWeightsReshapeKernel
port CLGEMMConvolutionLayer to ClGemmConvolution
Resolves: COMPMID-4515
Change-Id: I7d5b4ec72db2742f6eb9f3ffc88f717c35b4f2a3
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5983
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/gpu/cl/operators/ClGemm.h')
-rw-r--r-- | src/runtime/gpu/cl/operators/ClGemm.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/gpu/cl/operators/ClGemm.h b/src/runtime/gpu/cl/operators/ClGemm.h index aad208bdb0..254344e862 100644 --- a/src/runtime/gpu/cl/operators/ClGemm.h +++ b/src/runtime/gpu/cl/operators/ClGemm.h @@ -129,8 +129,8 @@ private: TensorInfo _tmp_b; bool _reshape_b_only_on_first_run; CLGEMMKernelType _gemm_kernel_type; - - experimental::MemoryRequirements _aux_mem{}; + bool _is_prepared; + experimental::MemoryRequirements _aux_mem{}; }; } // namespace opencl } // namespace arm_compute |