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

numba_dpex.dpctl_iface package

Submodules

  • numba_dpex.dpctl_iface.dpctl_capi_fn_builder module
    • DpctlCAPIFnBuilder
      • DpctlCAPIFnBuilder.get_dpctl_event_delete()
      • DpctlCAPIFnBuilder.get_dpctl_event_wait()
      • DpctlCAPIFnBuilder.get_dpctl_free_with_queue()
      • DpctlCAPIFnBuilder.get_dpctl_malloc_shared()
      • DpctlCAPIFnBuilder.get_dpctl_queue_delete()
      • DpctlCAPIFnBuilder.get_dpctl_queue_memcpy()
      • DpctlCAPIFnBuilder.get_dpctl_queue_submit_range()
      • DpctlCAPIFnBuilder.get_dpctl_queue_wait()
      • DpctlCAPIFnBuilder.get_dpctl_queuemgr_get_current_queue()
  • numba_dpex.dpctl_iface.dpctl_function_types module
    • dpctl_event_delete()
    • dpctl_event_wait()
    • dpctl_free_with_queue()
    • dpctl_get_current_queue()
    • dpctl_malloc_shared()
    • dpctl_queue_memcpy()
    • dpctl_queue_wait()
  • numba_dpex.dpctl_iface.legacy_kernel_launch_ops module
    • KernelLaunchOps
      • KernelLaunchOps.allocate_kernel_arg_array()
      • KernelLaunchOps.enqueue_kernel_and_copy_back()
      • KernelLaunchOps.free_queue()
      • KernelLaunchOps.get_current_queue()
      • KernelLaunchOps.process_kernel_arg()

Module contents

The dpctl_iface module implements Numba’s interface to the dpctl library that provides Python and C bindings to DPC++’s SYCL runtime API. The module includes:

  • LLVM IR builders for dpctl C API functions to be called directly from a Numba generated LLVM module.

  • Functions to launch kernels on the dpctl “current queue”.


© Copyright 2021-2023 Intel Corporation.

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