Trait Runtime

pub trait Runtime:
    Send
    + Sync
    + 'static
    + Debug {
    type Compiler: Compiler;
    type Server: ComputeServer<Kernel = Box<dyn CubeTask<Self::Compiler>>, Feature = Feature>;
    type Channel: ComputeChannel<Self::Server>;
    type Device: Default + Clone + Debug + Send + Sync;

    // Required methods
    fn device_id(device: &Self::Device) -> DeviceId;
    fn client(
        device: &Self::Device,
    ) -> ComputeClient<Self::Server, Self::Channel>;
    fn name(client: &ComputeClient<Self::Server, Self::Channel>) -> &'static str;
    fn supported_line_sizes() -> &'static [u8] ;
    fn max_cube_count() -> (u32, u32, u32);

    // Provided methods
    fn require_array_lengths() -> bool { ... }
    fn line_size_elem(elem: &Elem) -> impl Iterator<Item = u8> + Clone { ... }
}

Runtime for the CubeCL.

Required Associated Types

type Compiler: Compiler

The compiler used to compile the inner representation into tokens.

type Server: ComputeServer<Kernel = Box<dyn CubeTask<Self::Compiler>>, Feature = Feature>

The compute server used to run kernels and perform autotuning.

type Channel: ComputeChannel<Self::Server>

The channel used to communicate with the compute server.

type Device: Default + Clone + Debug + Send + Sync

The device used to retrieve the compute client.

Required Methods

fn device_id(device: &Self::Device) -> DeviceId

Fetch the id for the given device.

fn client(device: &Self::Device) -> ComputeClient<Self::Server, Self::Channel>

Retrieve the compute client from the runtime device.

fn name(client: &ComputeClient<Self::Server, Self::Channel>) -> &'static str

The runtime name on the given device.

fn supported_line_sizes() -> &'static [u8]

Returns the supported line sizes for the current runtime’s compiler.

fn max_cube_count() -> (u32, u32, u32)

Returns the maximum cube count on each dimension that can be launched.

Provided Methods

fn require_array_lengths() -> bool

Return true if global input array lengths should be added to kernel info.

fn line_size_elem(elem: &Elem) -> impl Iterator<Item = u8> + Clone

Returns all line sizes that are useful to perform IO operation on the given element.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors