ulib 0.3.3

Universal data storage library for CPU/GPU heterogeneous applications
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
64
65
66

use ulib::{ UVec, Device };

#[cfg(feature = "cuda")]
use ulib::{ AsCUDASlice, AsCUDASliceMut };

#[test]
fn test_simple_cpu() {
    let mut uvt: UVec<i32> = UVec::new_zeroed(100, Device::CPU);
    uvt.as_mut()[0] = 10;
    assert_eq!(uvt.as_mut()[2], 0);
    assert_eq!(uvt.as_mut()[0], 10);
    assert_eq!(uvt.clone().as_mut()[0], 10);
}

#[test]
fn test_realloc_cpu() {
    let mut uvt: UVec<i32> = UVec::new_zeroed(2, Device::CPU);
    uvt.as_mut()[0] = 10;
    unsafe { uvt.resize_uninit_preserve(3, Device::CPU); }
    assert!(uvt.len() == 3);
    assert!(uvt.capacity() >= 3);
    assert_eq!(uvt.as_ref()[0], 10);
    assert_eq!(uvt.as_ref()[1], 0);
    assert_eq!(uvt.clone().as_ref()[1], 0);
}

#[cfg(feature = "cuda")]
#[test]
fn test_simple_cuda() {
    let mut cuvt: UVec<i32> = UVec::new_zeroed(100, Device::CUDA(0));
    cuvt.as_mut();
    for i in 0..*ulib::NUM_CUDA_DEVICES {
        cuvt.as_cuda_slice(Device::CUDA(i as u8));
    }
    for i in 0..*ulib::NUM_CUDA_DEVICES {
        cuvt.as_cuda_slice_mut(Device::CUDA(i as u8));
    }
    for i in 0..*ulib::NUM_CUDA_DEVICES {
        cuvt.as_cuda_slice(Device::CUDA(i as u8));
    }
}

#[cfg(feature = "cuda")]
#[test]
fn test_realloc_cuda() {
    let mut uvt: UVec<i32> = UVec::new_zeroed(2, Device::CPU);
    uvt.as_mut()[0] = 10;
    unsafe { uvt.resize_uninit_preserve(3, Device::CUDA(0)); }
    assert!(uvt.len() == 3);
    assert!(uvt.capacity() >= 3);
    assert_eq!(uvt.as_ref()[0], 10);  // back to cpu.
    assert_eq!(uvt.as_ref()[1], 0);
    unsafe { uvt.resize_uninit_nopreserve(100, Device::CUDA(0)); }
    assert!(uvt.len() == 100);
    assert!(uvt.capacity() >= 100);
}

#[test]
fn test_vec_into() {
    let uvt = vec![7, 8, 9, 10];
    let uvt: UVec<u32> = uvt.into();
    assert_eq!(uvt.len(), 4);
    assert_eq!(uvt.capacity(), 4);
    assert_eq!(uvt.as_ref()[3], 10);
    assert_eq!(uvt.as_ref()[0], 7);
}