Struct CPU

pub struct CPU {
    pub cache: RefCell<Cache<RawCpuBuf>>,
    pub graph: RefCell<Graph>,
}

A CPU is used to perform calculations on the host CPU. To make new operations invocable, a trait providing new functions should be implemented for CPU.

Example

use custos::{CPU, VecRead, Buffer};

let device = CPU::new();
let a = Buffer::from((&device, [1, 2, 3]));

let out = device.read(&a);

assert_eq!(out, vec![1, 2, 3]);

Fields

cache: RefCell<Cache<RawCpuBuf>>

graph: RefCell<Graph>

Implementations

impl CPU

pub fn new() -> CPU

Creates an CPU with an InternCPU that holds an empty vector of pointers.

Trait Implementations

impl<T> ActivationOps<T> for CPU

where T: Float,

fn sigmoid(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn tanh(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn tanh_grad(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn relu(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn relu_grad(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

impl<'a, T: CDatatype + Float> AdamOp<'a, T> for CPU

fn step(&'a self, adam: &mut Adam<'a, T>, params: Vec<Param<'a, T>>)

impl<T> AdditionalOps<T> for CPU

where T: Number,

fn adds(&self, lhs: &Matrix<'_, T>, rhs: T) -> Matrix<'_, T>

fn muls(&self, lhs: &Matrix<'_, T>, rhs: T) -> Matrix<'_, T>

fn divs(&self, lhs: &Matrix<'_, T>, rhs: T) -> Matrix<'_, T>

impl<T> Alloc<T> for CPU

fn alloc(&self, len: usize) -> (*mut T, *mut c_void, u64)

Allocate memory on the implemented device.

fn with_data(&self, data: &[T]) -> (*mut T, *mut c_void, u64)

where T: Clone,

Allocate new memory with data

fn alloc_with_vec(&self, vec: Vec<T>) -> (*mut T, *mut c_void, u64)

If the vector vec was allocated previously, this function can be used in order to reduce the amount of allocations, which may be faster than using a slice of vec.

fn as_dev(&self) -> Device

Creates a generic representation of the device

impl AsDev for CPU

fn dev(&self) -> Device

where Self: Sized + Alloc<u8>,

impl<T> AssignOps<T> for CPU

where T: Number,

fn add_assign(&self, lhs: &mut Buffer<'_, T>, rhs: &Buffer<'_, T>)

Add assign

fn sub_assign(&self, lhs: &mut Buffer<'_, T>, rhs: &Buffer<'_, T>)

impl<T> BaseOps<T> for CPU

where T: Number,

fn add(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

Element-wise addition

fn sub(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

Element-wise subtraction

fn mul(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

Element-wise multiplication

fn div(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

Element-wise division

impl<'a, T> CacheBuf<'a, T> for CPU

fn cached(&'a self, len: usize) -> Buffer<'a, T>

Adds a buffer to the cache. Following calls will return this buffer, if the corresponding internal count matches with the id used in the cache.

impl CacheReturn<RawCpuBuf> for CPU

fn cache(&self) -> RefMut<'_, Cache<RawCpuBuf>>

Returns a device specific Cache.

impl<T> ClearBuf<T> for CPU

where T: Default,

fn clear(&self, buf: &mut Buffer<'_, T>)

Sets all elements of the matrix to zero.

impl<T> ClipOp<T> for CPU

where T: Number,

fn clip(&self, x: &Matrix<'_, T>, min: T, max: T) -> Matrix<'_, T>

impl<'a, T> CloneBuf<'a, T> for CPU

where T: Clone,

fn clone_buf(&'a self, buf: &Buffer<'a, T>) -> Buffer<'a, T>

Creates a deep copy of the specified buffer.

impl<T> ColOp<T> for CPU

where T: Number,

fn add_col(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

fn sub_col(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

fn div_col(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

impl Debug for CPU

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for CPU

fn default() -> CPU

Returns the “default value” for a type.

impl<T> DiagflatOp<T> for CPU

where T: Default + Copy,

fn diagflat(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T> FnsOps<T> for CPU

where T: Float,

fn exp(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn ln(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn neg(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn powf(&self, x: &Matrix<'_, T>, rhs: T) -> Matrix<'_, T>

fn powi(&self, x: &Matrix<'_, T>, rhs: i32) -> Matrix<'_, T>

impl<T> Gemm<T> for CPU

where T: GenericBlas + Default + Copy,

fn gemm(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

impl GraphReturn for CPU

fn graph(&self) -> RefMut<'_, Graph>

impl<T> MaxOps<T> for CPU

where T: Number,

fn max(&self, x: &Matrix<'_, T>) -> T

fn max_rows(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn max_cols(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T: Number> OnehotOp<T> for CPU

fn onehot(&self, matrix: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T> RandOp<T> for CPU

where T: Float,

fn rand(&self, x: &mut Buffer<'_, T>, lo: T, hi: T)

impl<T> RowOp<T> for CPU

where T: Number,

fn add_row(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

fn add_row_mut(&self, lhs: &mut Matrix<'_, T>, rhs: &Matrix<'_, T>)

impl<T: CDatatype> SGDOp<T> for CPU

fn step_momentum(&self, sgd: &mut SGD<'_, T>, params: Vec<Param<'_, T>>)

fn step(&self, sgd: &mut SGD<'_, T>, params: Vec<Param<'_, T>>)

impl<T> ScalarAssign<T> for CPU

where T: Copy + AddAssign + MulAssign + DivAssign + SubAssign,

fn adds_assign(&self, lhs: &mut Matrix<'_, T>, rhs: T)

fn muls_assign(&self, lhs: &mut Matrix<'_, T>, rhs: T)

fn divs_assign(&self, lhs: &mut Matrix<'_, T>, rhs: T)

fn subs_assign(&self, lhs: &mut Matrix<'_, T>, rhs: T)

impl<T> SoftmaxOps<T> for CPU

where T: Float + GenericBlas,

fn softmax(&self, inputs: &Matrix<'_, T>) -> Matrix<'_, T>

fn softmax_grad( &self, activated: &Matrix<'_, T>, grads: &Matrix<'_, T>, ) -> Matrix<'_, T>

impl<T> SumOps<T> for CPU

where T: Number,

fn sum(&self, x: &Matrix<'_, T>) -> T

fn mean(&self, x: &Matrix<'_, T>) -> T

fn sum_rows(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

fn sum_cols(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T> TransposeOp<T> for CPU

where T: Default + Copy,

fn transpose(&self, x: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T> VecRead<T> for CPU

where T: Clone,

fn read(&self, buf: &Buffer<'_, T>) -> Vec<T>

Read the data of a buffer into a vector

impl<T> WriteBuf<T> for CPU

where T: Copy,

fn write(&self, buf: &mut Buffer<'_, T>, data: &[T])

Write data to the buffer.

fn write_buf(&self, _dst: &mut Buffer<'_, T>, _src: &Buffer<'_, T>)

Writes data from Buffer to other Buffer.

impl<T> CCEOp<T> for CPU

where T: Float + CDatatype,