Contents

class

SushiAI::Graph::DeviceFusionSelector

Selects fused forms based on device profile and execution mode.

Declared in
include/SushiAI/graph/fusion_selector.hpp
Inherits
SushiAI::Graph::IFusionSelector

Public member functions

DeviceFusionSelector(const SushiBLAS::Device::Profile &profile, SushiBLAS::ExecutionMode mode) noexcept

Constructs a selector with explicit profile and execution mode.

Parameters

profile

The device profile.

mode

The engine execution mode.

const SushiBLAS::Device::Profile & profile() const noexcept

The device this selector is reasoning about.

SushiBLAS::ExecutionMode mode() const noexcept

The execution mode this selector is reasoning about.

virtual FusionChoice choose(const FusionQuery &query) const override

Chooses the form to spell for query.

Parameters

query

The matched node.

Returns

The chosen form and the reason for it.

Static public member functions

static DeviceFusionSelector for_engine(SushiBLAS::Engine &engine)

Builds a selector describing engine's real device.

Parameters

engine

The engine the step will be recorded on.

Returns

A selector over that device's cached profile.