Struct GpuStore

pub struct GpuStore { /* private fields */ }

Implementations

impl GpuStore

pub fn get_default() -> &'static GpuInstance

Examples found in repository

examples/assignment/main.rs (line 4)

fn main() {
    println!("Running in {:?}", GpuStore::get_default().info());
    let mut ma = Tensor::from_data_and_shape(vec![1., 2., 3., 4., 5., 6., 7., 8.], vec![2, 2, 2]);
    println!("{:?}", ma);
    ma.assign(s!(1..2; ..; 1..2), 10.);
    ma.assign(s!(0;0;0), -50.);
    println!("{:?}", ma);
    /*
    Shape: [2, 2, 2]
    [[[ -50  2 ]
      [ 3  4 ]]

     [[ 5  10 ]
      [ 7  10 ]]]
    */
}

examples/simple_mm/main.rs (line 4)

fn main() {
    println!("Running in {:?}", GpuStore::get_default().info());
    let ma = Tensor::from_data_and_shape(vec![1., 2., 3., 4., 5., 6., 7., 8.], vec![2, 2, 2]);
    let mb = Tensor::from_data_and_shape(vec![2., 3., 4., 5.], vec![2, 2]);
    let result = ma.matmul(&mb);
    println!("{:?}", result);
    /*
    Running in AdapterInfo { name: "AMD Radeon Pro 560", vendor: 0, device: 0, device_type: DiscreteGpu, backend: Metal }
    Shape: [2, 2, 2]
    [[[ 10  13 ]
      [ 22  29 ]]

     [[ 34  45 ]
      [ 46  61 ]]]
    */
}

pub fn get(gpu_info: &AdapterInfo) -> &'static GpuInstance

pub async fn select_gpu(gpu_info: &AdapterInfo)

pub fn list_gpus() -> Vec<&'static AdapterInfo>