xlog-cuda 0.9.2

CUDA kernel provider, buffers, and interop for XLOG
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
67
68
69
70
71
72
73
74
75
76
77
78
79

mod common;
use common::setup_provider;
use xlog_core::{ScalarType, Schema};

#[test]
fn test_pack_keys_gpu_generic_no_dtoh() {
    let Some(provider) = setup_provider() else {
        eprintln!("Skipping: no CUDA device");
        return;
    };

    let schema = Schema::new(vec![
        ("a".to_string(), ScalarType::U32),
        ("b".to_string(), ScalarType::U32),
        ("c".to_string(), ScalarType::U32),
        ("d".to_string(), ScalarType::U32),
        ("e".to_string(), ScalarType::U32),
    ]);

    let a = vec![1u32, 2, 3];
    let b = vec![4u32, 5, 6];
    let c = vec![7u32, 8, 9];
    let d = vec![10u32, 11, 12];
    let e = vec![13u32, 14, 15];

    let buffer = provider
        .create_buffer_from_u32_columns(&[&a, &b, &c, &d, &e], schema)
        .unwrap();

    provider.reset_host_transfer_stats();

    let index = provider
        .build_join_index_v2(&buffer, &[0, 1, 2, 3, 4])
        .unwrap();
    assert!(index.estimated_bytes() > 0);

    let stats = provider.host_transfer_stats();
    assert_eq!(
        stats.dtoh_bytes, 0,
        "unexpected device-to-host transfers during GPU key packing: {} bytes",
        stats.dtoh_bytes
    );
}

#[test]
fn test_pack_keys_gpu_common_path_no_host_transfers() {
    let Some(provider) = setup_provider() else {
        eprintln!("Skipping: no CUDA device");
        return;
    };

    let schema = Schema::new(vec![
        ("a".to_string(), ScalarType::U32),
        ("b".to_string(), ScalarType::U32),
    ]);

    let a = vec![1u32, 2, 3, 4];
    let b = vec![10u32, 20, 30, 40];
    let buffer = provider
        .create_buffer_from_u32_columns(&[&a, &b], schema)
        .unwrap();

    provider.reset_host_transfer_stats();

    let index = provider.build_join_index_v2(&buffer, &[0, 1]).unwrap();
    assert!(index.estimated_bytes() > 0);

    let stats = provider.host_transfer_stats();
    assert_eq!(
        stats.dtoh_bytes, 0,
        "unexpected device-to-host transfer during common GPU key packing: {:?}",
        stats
    );
    assert_eq!(
        stats.htod_bytes, 0,
        "unexpected host-to-device transfer during common GPU key packing: {:?}",
        stats
    );
}