diff options
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
| -rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 6ebdf298f1..94cc02a705 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -58,6 +58,19 @@ void CLBuildOptions::add_option_if_else(bool cond, std::string option_true, std: (cond) ? add_option(std::move(option_true)) : add_option(std::move(option_false)); } +void CLBuildOptions::add_options(const StringSet &options) +{ + _build_opts.insert(options.begin(), options.end()); +} + +void CLBuildOptions::add_options_if(bool cond, const StringSet &options) +{ + if(cond) + { + add_options(options); + } +} + const CLBuildOptions::StringSet &CLBuildOptions::options() const { return _build_opts; @@ -299,8 +312,11 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "sobel_separable7x1", "sobel_filter.cl" }, { "sobel_separable1x7", "sobel_filter.cl" }, { "softmax_layer_max", "softmax_layer.cl" }, + { "softmax_layer_max_quantized", "softmax_layer_quantized.cl" }, { "softmax_layer_shift_exp_sum", "softmax_layer.cl" }, + { "softmax_layer_shift_exp_sum_quantized", "softmax_layer_quantized.cl" }, { "softmax_layer_norm", "softmax_layer.cl" }, + { "softmax_layer_norm_quantized", "softmax_layer_quantized.cl" }, { "softmax_layer_max_shift_exp_sum_serial", "softmax_layer.cl" }, { "softmax_layer_max_shift_exp_sum_parallel", "softmax_layer.cl" }, { "suppress_non_maximum", "canny.cl" }, @@ -587,6 +603,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map = #include "./cl_kernels/softmax_layer.clembed" }, { + "softmax_layer_quantized.cl", +#include "./cl_kernels/softmax_layer_quantized.clembed" + }, + { "tablelookup.cl", #include "./cl_kernels/tablelookup.clembed" }, |