numba_dpex.core.parfors.reduction_helper module

class numba_dpex.core.parfors.reduction_helper.ReductionHelper

Bases: object

The class to define and allocate reduction intermediate variables.

class numba_dpex.core.parfors.reduction_helper.ReductionKernelVariables(lowerer, parfor_node, typemap, parfor_outputs, reductionHelperList)

Bases: object

The parfor body and the main function body share ir.Var nodes. We have to do some replacements of Var names in the parfor body to make them legal parameter names. If we don’t copy then the Vars in the main function also would incorrectly change their name.

copy_final_sum_to_host(psrfor_kernel)
property final_sum_names
property func_arg_types
property ind_dict
property legal_loop_indices
property loop_body
property lowerer
property param_dict
property param_types
property parfor_legalized_params
property parfor_reddict
property parfor_redvars
property parfor_redvars_to_redarrs
property work_group_size