vkml 0.0.3

High-level Vulkan-based machine learning library
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

use std::cell::UnsafeCell;

use crate::tensor::Tensor;

pub struct TensorCell {
    tensor: UnsafeCell<Tensor>,
}

unsafe impl Sync for TensorCell {}

impl TensorCell {
    pub fn new(t: Tensor) -> Self {
        Self {
            tensor: UnsafeCell::new(t),
        }
    }

    pub unsafe fn as_ref(&self) -> &Tensor {
        unsafe { &*self.tensor.get() }
    }

    // safety: returns mutable ref via UnsafeCell; caller must ensure exclusive access
    #[allow(clippy::mut_from_ref)]
    pub unsafe fn as_mut(&self) -> &mut Tensor {
        unsafe { &mut *self.tensor.get() }
    }
}