vortx 0.2.0

Cross-platform GPU tensor library with Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! Fundamental linear-algebra matrix/vector operations.

mod contiguous;
mod gemm;
mod op_assign;
mod reduce;
mod repeat;

pub use contiguous::Contiguous;
pub use gemm::{Gemm, MatrixMode, N, T};
pub use op_assign::{BinOpOffsets, OpAssign, OpAssignVariant};
pub use reduce::{Reduce, ReduceVariant};
pub use repeat::Repeat;

/// Returns the components of an nalgebra matrix (column-major) as a row-major buffer.
#[cfg(test)]
pub fn to_row_major(mat: &nalgebra::DMatrix<f32>) -> Vec<f32> {
    mat.transpose().as_slice().to_vec()
}