ruda-kernel 0.1.4

Ruda Rust kernel DSL and device tensor operations.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use ruda_core::{kernel::KernelDefinition, launch::ExecutionMode};
use ruda_compiler::optimizer::{Optimizer, OptimizerBuilder};
use crate::dsl::post_processing::{checked_io::CheckedIoProcessor, predicate::PredicateProcessor, saturating::SaturatingArithmeticProcessor};

mod erf_transform;
mod trigonometries_transform;
use erf_transform::ErfTransform;
use trigonometries_transform::{HypotTransform, RhypotTransform};

pub fn optimize(kernel: &KernelDefinition, mode: ExecutionMode) -> Optimizer {
    OptimizerBuilder::default()
            .with_transformer(ErfTransform)
            .with_transformer(HypotTransform)
            .with_transformer(RhypotTransform)
            .with_processor(CheckedIoProcessor::new(
                mode,
                kernel.options.kernel_name.clone(),
            ))
            .with_processor(SaturatingArithmeticProcessor::new(true))
            .with_processor(PredicateProcessor)
            .optimize(kernel.body.clone(), kernel.ruda_dim)
}