rds 0.0.3

Rust Data Science
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

#[allow(unused_imports)]
use array::NDArray;
#[allow(unused_imports)]
use backend;
#[allow(unused_imports)]
use backend::compute_backend::{SizedBuffer, ComputeBackend};

#[test]
#[cfg(feature = "opencl")]
fn opencl_enumerate_devices() {
    let platforms = backend::opencl::OpenCL::get_platform_list().unwrap();
    for platform in platforms {
        println!("{}", platform);
        let devices = backend::opencl::OpenCL::get_device_list(&platform).unwrap();
        for device in devices {
            println!("{}", device);
        }
    }
}

#[test]
#[cfg(feature = "opencl")]
fn opencl_init_and_finalize_devices() {
    let platforms = backend::opencl::OpenCL::get_platform_list().unwrap();
    for platform in platforms {
        println!("{}", platform);
        let mut devices = backend::opencl::OpenCL::get_device_list(&platform).unwrap();
        for device in &mut devices {
            device.init().unwrap();
            device.finalize().unwrap();
        }
    }
}

#[test]
#[cfg(feature = "opencl")]
fn opencl_backend_array() {
    let mut array1 = NDArray::<f32>::new(&[3,3], 0f32); 
    let mut array2 = NDArray::<f32>::new(&[3,3], 0f32); 
    for i in 0..3 {
        for j in 0..3 {
            array1[&[i,j]] = (i * 3 + j * 5) as f32;
        }
    }

    let platforms = backend::opencl::OpenCL::get_platform_list().unwrap();
    for platform in platforms {
        let mut devices = backend::opencl::OpenCL::get_device_list(&platform).unwrap();
        for device in &mut devices {
            device.init().unwrap();
            let id = device.create_array(array1.get_raw_size()).unwrap();       
            device.set_array(id, &array1).unwrap();
            device.get_array(id, &mut array2).unwrap();
            for i in 0..3 {
                for j in 0..3 {
                    assert_eq!(array2[&[i,j]], (i * 3 + j * 5) as f32);
                }
            }
            device.delete_array(id).unwrap();
            device.finalize().unwrap();
        }
    }
}