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

use crate::dma::buffer::DmaBuffer;
use crate::tpu::tensor::Tensor;

pub fn alloc_tensor(len: usize, align: usize) -> Option<Tensor> {
    let db = DmaBuffer::new(len, align)?;
    Some(Tensor::new_from_buffer([len, 1, 1, 1], db, len))
}

pub fn size_of(t: &Tensor) -> usize {
    t.len
}

pub struct TpuMemory;

impl TpuMemory {
    pub fn alloc(size: usize, align: usize) -> Option<DmaBuffer> {
        DmaBuffer::new(size, align)
    }
}