simple_mm/
main.rs

1use tensor_compute::{GpuStore, Tensor};
2
3fn main() {
4    println!("Running in {:?}", GpuStore::get_default().info());
5    let ma = Tensor::from_data_and_shape(vec![1., 2., 3., 4., 5., 6., 7., 8.], vec![2, 2, 2]);
6    let mb = Tensor::from_data_and_shape(vec![2., 3., 4., 5.], vec![2, 2]);
7    let result = ma.matmul(&mb);
8    println!("{:?}", result);
9    /*
10    Running in AdapterInfo { name: "AMD Radeon Pro 560", vendor: 0, device: 0, device_type: DiscreteGpu, backend: Metal }
11    Shape: [2, 2, 2]
12    [[[ 10  13 ]
13      [ 22  29 ]]
14
15     [[ 34  45 ]
16      [ 46  61 ]]]
17    */
18}
simple_mm/main.rs

simple_mm/
main.rs
