ruDNN 0.2.6

Neural network operators for Ruda.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

use ruda_kernel::dsl::prelude::*;
use ruda_kernel::dsl as kernel_dsl;

use rublas::kernel_ir::components::global::{GlobalWriterConfig, PartitionedStage, WriteEventListener};

mod plane;
mod unit;

use ruda_kernel::library::tensor::View;
use ruda_kernel::library::tensor::layout::Coords2d;
pub use plane::*;
pub use unit::*;

use crate::attention::kernel_ir::components::stage::StageAttentionConfig;

#[ruda]
pub trait AttentionWriter<ES: Numeric, ESS: Size, EG: Numeric, EGS: Size>:
    WriteEventListener
{
    fn init<S: StageAttentionConfig>(
        global: View<Vector<EG, EGS>, Coords2d, ReadWrite>,
        #[comptime] config: GlobalWriterConfig,
    ) -> Self;

    fn stage(&mut self) -> PartitionedStage<ES, ESS>;
}