hardware 0.0.9

A no_std bare-metal hardware abstraction layer — all port I/O, memory and swap allocations are guarded at runtime. Do not consider this dependency stable before x.1.x
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
27
28
29
30
31
32

use crate::dma::buffer::DmaBuffer;

pub struct Tensor {
    pub dims: [usize; 4],
    buf: Option<DmaBuffer>,
    pub len: usize,
}

impl Tensor {
    pub fn new_from_buffer(dims: [usize; 4], buf: DmaBuffer, len: usize) -> Self {
        Tensor {
            dims,
            buf: Some(buf),
            len,
        }
    }

    pub fn as_ptr(&self) -> *mut u8 {
        self.buf
            .as_ref()
            .map(|b| b.as_ptr())
            .unwrap_or(core::ptr::null_mut())
    }

    pub fn numel(&self) -> usize {
        self.dims.iter().product()
    }

    pub fn take_buffer(&mut self) -> Option<DmaBuffer> {
        core::mem::take(&mut self.buf)
    }
}