diff options
Diffstat (limited to 'src/core/CL/CLHelpers.cpp')
-rw-r--r-- | src/core/CL/CLHelpers.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/core/CL/CLHelpers.cpp b/src/core/CL/CLHelpers.cpp index 3323929742..5c53455eeb 100644 --- a/src/core/CL/CLHelpers.cpp +++ b/src/core/CL/CLHelpers.cpp @@ -466,4 +466,16 @@ bool export_weights_to_cl_image(const ITensorInfo *tensor) return true; } +void set_unroll_with_pragma(CLBuildOptions &built_opts, std::initializer_list<int> values) +{ + for(const int value : values) + { + if(value > max_manual_loop_unrolling) + { + built_opts.add_option("-DUNROLL_WITH_PRAGMA"); + return; + } + } +} + } // namespace arm_compute |