rustacuda 0.1.3

CUDA Driver API Wrapper
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

extern crate rustacuda;

use rustacuda::prelude::*;
use rustacuda::quick_init;
use std::sync::mpsc::sync_channel;

#[test]
fn test_stream_callbacks_execution_order() {
    let _ctx = quick_init();
    let stream = Stream::new(StreamFlags::NON_BLOCKING, None).unwrap();

    let (order_sender, order_receiver) = sync_channel(0);
    stream
        .add_callback(Box::new(|_| {
            order_sender.send(1).unwrap();
        }))
        .unwrap();
    stream
        .add_callback(Box::new(|_| {
            order_sender.send(2).unwrap();
        }))
        .unwrap();
    stream
        .add_callback(Box::new(|_| {
            order_sender.send(3).unwrap();
        }))
        .unwrap();
    for expected in &[1, 2, 3] {
        assert_eq!(*expected, order_receiver.recv().unwrap());
    }
}

#[test]
fn test_stream_callbacks_environment_capture() {
    let _ctx = quick_init();
    let stream = Stream::new(StreamFlags::NON_BLOCKING, None).unwrap();

    let (capture_sender, capture_receiver) = sync_channel(0);
    let magic_numbers = (42, Box::new(1337));
    stream
        .add_callback(Box::new(|_| {
            capture_sender.send(magic_numbers).unwrap();
        }))
        .unwrap();
    let captured_magic_numbers = capture_receiver.recv().unwrap();
    assert_eq!(42, captured_magic_numbers.0);
    assert_eq!(1337, *captured_magic_numbers.1);
}

#[test]
fn test_stream_callbacks_status_propagation() {
    let _ctx = quick_init();
    let stream = Stream::new(StreamFlags::NON_BLOCKING, None).unwrap();

    let (status_sender, status_receiver) = sync_channel(0);
    stream
        .add_callback(Box::new(|status| {
            status_sender.send(status).unwrap();
        }))
        .unwrap();
    assert_eq!(Ok(()), status_receiver.recv().unwrap())
}