numba_dpex.core.utils.kernel_templates package

Submodules

• numba_dpex.core.utils.kernel_templates.kernel_template_iface module
  • KernelTemplateInterface
    • KernelTemplateInterface.dump_kernel_ir()
    • KernelTemplateInterface.dump_kernel_string()
    • KernelTemplateInterface.kernel_ir
    • KernelTemplateInterface.kernel_string
• numba_dpex.core.utils.kernel_templates.range_kernel_template module
  • RangeKernelTemplate
    • RangeKernelTemplate.dump_kernel_ir()
    • RangeKernelTemplate.dump_kernel_string()
    • RangeKernelTemplate.kernel_ir
    • RangeKernelTemplate.kernel_string
• numba_dpex.core.utils.kernel_templates.reduction_template module
  • RemainderReduceIntermediateKernelTemplate
    • RemainderReduceIntermediateKernelTemplate.dump_kernel_ir()
    • RemainderReduceIntermediateKernelTemplate.dump_kernel_string()
    • RemainderReduceIntermediateKernelTemplate.kernel_ir
    • RemainderReduceIntermediateKernelTemplate.kernel_string
  • TreeReduceIntermediateKernelTemplate
    • TreeReduceIntermediateKernelTemplate.dump_kernel_ir()
    • TreeReduceIntermediateKernelTemplate.dump_kernel_string()
    • TreeReduceIntermediateKernelTemplate.kernel_ir
    • TreeReduceIntermediateKernelTemplate.kernel_string

Module contents

class numba_dpex.core.utils.kernel_templates.RangeKernelTemplate(kernel_name, kernel_params, kernel_rank, ivar_names, sentinel_name, loop_ranges, param_dict)

Bases: object

A template class to generate a numba_dpex.kernel decorated function representing a basic range kernel.

dump_kernel_ir()

Helper to dump the Numba IR for the RangeKernelTemplate.

dump_kernel_string()

Helper to print the kernel function string.

property kernel_ir

Returns the Numba IR generated for a RangeKernelTemplate.

Returns: The Numba functionIR object for the compiled kernel_txt string.

property kernel_string

Returns the function string generated for a RangeKernelTemplate.

Returns:

str: A string representing a stub kernel function for the parfor.