Struct KernelFusionEngine 

pub struct KernelFusionEngine {
    pub patterns: Vec<FusionPattern>,
    pub constraints: Vec<FusionConstraint>,
    pub generated_kernels: Arc<RwLock<HashMap<String, FusedKernel>>>,
    pub performance_database: Arc<RwLock<PerformanceDatabase>>,
    pub fusion_statistics: Arc<RwLock<FusionStatistics>>,
}

Kernel fusion engine

Fields

patterns: Vec<FusionPattern>

constraints: Vec<FusionConstraint>

generated_kernels: Arc<RwLock<HashMap<String, FusedKernel>>>

performance_database: Arc<RwLock<PerformanceDatabase>>

fusion_statistics: Arc<RwLock<FusionStatistics>>

Implementations

impl KernelFusionEngine

pub fn new() -> Self

pub fn analyze_graph( &self, graph: &ComputationGraph, ) -> Result<Vec<FusionOpportunity>>

pub fn fuse_operations( &self, graph: &ComputationGraph, opportunity: &FusionOpportunity, ) -> Result<FusedKernel>

pub fn shapes_broadcastable(&self, shape1: &[usize], shape2: &[usize]) -> bool

Trait Implementations

impl Default for KernelFusionEngine

fn default() -> Self

Returns the “default value” for a type.