ex_matrix.rs

Add two device matrices. Create, add, and explicitly read GPU matrices.

ExSDK source-ownedRustCore

Source

Example

1use oa::{Engine, matrix};
2
3fn main() -> oa::Result<()> {
4 let engine = Engine::new()?;
5
6 let one = matrix::ones(&engine, [2, 3])?;
7 let two = matrix::full(&engine, [2, 3], 2.0)?;
8 let result = matrix::add(&one, &two)?;
9
10 let values = result.read_f32()?;
11 assert_eq!(values.len(), 6);
12 assert!(values.iter().all(|&v| (v - 3.0).abs() <= 1e-6));
13
14 println!("Matrix addition verified: every value is 3");
15 Ok(())
16}
17