numba-dpex

Core Features

  • Code-generation based on a device
  • Automatic offload of NumPy expressions

User Guides

  • Getting Started
  • Direct kernel programming
  • Debugging with GDB
  • Docker
  • numba-dpex for numba.cuda Programmers

Developer Guides

  • dpnp integration
  • Debugging the compilation pipeline
numba-dpex
  • numba_dpex.core.parfors package
  • View page source

numba_dpex.core.parfors package

Submodules

  • numba_dpex.core.parfors.kernel_builder module
    • ParforKernel
    • create_kernel_for_parfor()
    • update_sentinel()
  • numba_dpex.core.parfors.parfor_lowerer module
    • ParforLowerFactory
      • ParforLowerFactory.device_to_lowerer_map
      • ParforLowerFactory.get_lowerer()
    • ParforLowerImpl
  • numba_dpex.core.parfors.reduction_helper module
    • ReductionHelper
    • ReductionKernelVariables
      • ReductionKernelVariables.copy_final_sum_to_host()
      • ReductionKernelVariables.final_sum_names
      • ReductionKernelVariables.func_arg_types
      • ReductionKernelVariables.ind_dict
      • ReductionKernelVariables.legal_loop_indices
      • ReductionKernelVariables.loop_body
      • ReductionKernelVariables.lowerer
      • ReductionKernelVariables.param_dict
      • ReductionKernelVariables.param_types
      • ReductionKernelVariables.parfor_legalized_params
      • ReductionKernelVariables.parfor_redarrs_legal_dict
      • ReductionKernelVariables.parfor_reddict
      • ReductionKernelVariables.parfor_redvars
      • ReductionKernelVariables.parfor_redvars_to_redarrs
      • ReductionKernelVariables.redvars_legal_dict
      • ReductionKernelVariables.work_group_size
  • numba_dpex.core.parfors.reduction_kernel_builder module
    • create_reduction_main_kernel_for_parfor()
    • create_reduction_remainder_kernel_for_parfor()

Module contents


© Copyright 2021-2023 Intel Corporation.

Built with Sphinx using a theme provided by Read the Docs.