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.kernel_interface package
  • View page source

numba_dpex.core.kernel_interface package

Submodules

  • numba_dpex.core.kernel_interface.arg_pack_unpacker module
    • Packer
      • Packer.unpacked_args
  • numba_dpex.core.kernel_interface.dispatcher module
    • JitKernel
      • JitKernel.cache
      • JitKernel.cache_hits
  • numba_dpex.core.kernel_interface.func module
    • DpexFunction
      • DpexFunction.compile()
    • DpexFunctionTemplate
      • DpexFunctionTemplate.cache
      • DpexFunctionTemplate.cache_hits
      • DpexFunctionTemplate.compile()
    • compile_func()
    • compile_func_template()
  • numba_dpex.core.kernel_interface.indexers module
    • NdRange
      • NdRange.get_global_range()
      • NdRange.get_local_range()
      • NdRange.global_range
      • NdRange.local_range
    • Range
      • Range.get()
      • Range.size()
  • numba_dpex.core.kernel_interface.kernel_base module
    • KernelInterface
      • KernelInterface.compile()
      • KernelInterface.device_driver_ir_module
      • KernelInterface.llvm_module
      • KernelInterface.module_name
      • KernelInterface.pyfunc_name
  • numba_dpex.core.kernel_interface.spirv_kernel module
    • SpirvKernel
      • SpirvKernel.compile()
      • SpirvKernel.device_driver_ir_module
      • SpirvKernel.llvm_module
      • SpirvKernel.module_name
      • SpirvKernel.pyfunc_name
      • SpirvKernel.target_context
      • SpirvKernel.typing_context
  • numba_dpex.core.kernel_interface.utils module
    • chk_compute_follows_data_compliance()
    • determine_kernel_launch_queue()

Module contents

Defines the interface for kernel compilation using numba-dpex.


© Copyright 2021-2023 Intel Corporation.

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