Crate rust_gpu_tools

source ·
Expand description

Abstraction layer for OpenCL and CUDA.

Environment variables

  • RUST_GPU_TOOLS_CUSTOM_GPU

    rust-gpu-tools has a hard-coded list of GPUs and their CUDA core count. If your card is not part of that list, you can add it via RUST_GPU_TOOLS_CUSTOM_GPU. The value is a comma separated list of name:cores. Example:

    RUST_GPU_TOOLS_CUSTOM_GPU="GeForce RTX 2080 Ti:4352,GeForce GTX 1060:1280"

Feature flags

There are two feature flags, cuda and opencl. By default opencl is enabled. You can enable both at the same time. At least one of them needs to be enabled at any time.

Modules

Macros

  • program_closures
    Creates two closures, one for CUDA, one for OpenCL for the given one.

Structs

  • Device
    A device that may have a CUDA and/or OpenCL GPU associated with it.
  • DeviceUuid
    A unique identifier based on UUID of the device.
  • LocalBuffer
    A buffer on the GPU.
  • PciId
    The PCI-ID is the combination of the PCI Bus ID and PCI Device ID.

Enums

  • Framework
    Which framework to use, CUDA or OpenCL.
  • GPUError
    Error types of this library.
  • Program
    Abstraction for running programs on CUDA or OpenCL.
  • UniqueId
    Unique identifier that can either be a PCI ID or a UUID.
  • Vendor
    Currently supported vendors of this library.

Statics