numba_dpex.core.parfors.kernel_templates.reduction_template

Overview

Classes

TreeReduceIntermediateKernelTemplate

The class to build reduction main kernel_txt template and

RemainderReduceIntermediateKernelTemplate

The class to build reduction remainder kernel_txt template and

Classes

class TreeReduceIntermediateKernelTemplate(kernel_name, kernel_params, ivar_names, sentinel_name, loop_ranges, param_dict, parfor_dim, redvars, parfor_args, parfor_reddict, redvars_dict, local_accessors_dict, typemap)

Bases: numba_dpex.core.parfors.kernel_templates.kernel_template_iface.KernelTemplateInterface

The class to build reduction main kernel_txt template and compiled Numba functionIR.

Overview

Methods

dump_kernel_string()

Helper to print the kernel function string.

Members

dump_kernel_string()

Helper to print the kernel function string.

class RemainderReduceIntermediateKernelTemplate(kernel_name, kernel_params, sentinel_name, redvars, parfor_reddict, redvars_dict, typemap, legal_loop_indices, global_size_var_name, global_size_mod_var_name, partial_sum_size_var_name, partial_sum_var_name, final_sum_var_name, reductionKernelVar)

Bases: numba_dpex.core.parfors.kernel_templates.kernel_template_iface.KernelTemplateInterface

The class to build reduction remainder kernel_txt template and compiled Numba functionIR.

Overview

Methods

dump_kernel_string()

Helper to print the kernel function string.

Members

dump_kernel_string()

Helper to print the kernel function string.