diff options
Diffstat (limited to 'examples/gemm_tuner/cl_gemmlowp_reshaped.cpp')
| -rw-r--r-- | examples/gemm_tuner/cl_gemmlowp_reshaped.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/examples/gemm_tuner/cl_gemmlowp_reshaped.cpp b/examples/gemm_tuner/cl_gemmlowp_reshaped.cpp index 8d100f613a..7c26e0f389 100644 --- a/examples/gemm_tuner/cl_gemmlowp_reshaped.cpp +++ b/examples/gemm_tuner/cl_gemmlowp_reshaped.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020 Arm Limited. + * Copyright (c) 2020-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -244,7 +244,11 @@ public: if(rhs_info.export_to_cl_image) { - examples::gemm_tuner_helpers::update_padding_for_cl_image(rhs_reshaped.info()); + if(!examples::gemm_tuner_helpers::update_padding_for_cl_image(rhs_reshaped.info())) + { + std::cerr << "cl_image is not supported on the device, disable export_to_cl_image" << std::endl; + return false; + } } GEMMReshapeInfo gemm_info |