custos-math

This crate provides CUDA, OpenCL and CPU based matrix operations using [custos].

Installation

Add "custos-math" as a dependency: You will also need [custos], if you want to run an example. ```toml [dependencies] custos-math = "0.4.6" custos = "0.4.6"

to disable the default features (cuda, opencl) and use an own set of features:

custos-math = {version = "0.4.6", default-features=false, features=["opencl"]}

custos = { version="0.4.6", default-features = false, features=["opencl"]}

```

Example

```rust use custos::CPU; use custos_math::Matrix;

fn main() { let device = CPU::new();

let a = Matrix::from((&device, (2, 3), [1., 2., 3., 4., 5., 6.,]));
let b = Matrix::from((&device, (3, 2), [6., 5., 4., 3., 2., 1.,]));

let c = a.gemm(&b);

assert_eq!(c.read(), vec![20., 14., 56., 41.,]);

} ```

Many more examples can be found in the tests and examples folder.