vyre-driver-cuda 0.6.1

CUDA/PTX backend for vyre through the CUDA driver API.
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

//! Parity test: vyre-primitives predicate node_kind_eq + literal_of
//! match CPU oracles.

#![cfg(test)]

mod common;

use common::{cuda_u32_bitset_output, with_live_backend};
use vyre_primitives::predicate::literal_of::{cpu_ref as literal_cpu, literal_of};
use vyre_primitives::predicate::node_kind_eq::{cpu_ref as kind_eq_cpu, node_kind_eq};

fn run_node_kind_eq(nodes: &[u32], kind: u32) -> Vec<u32> {
    let n = nodes.len() as u32;
    let program = node_kind_eq("nodes", "nodeset", n, kind);
    with_live_backend("node kind predicate", |backend| {
        cuda_u32_bitset_output(backend, &program, n, nodes, "node_kind_eq")
    })
}

#[test]
fn cuda_node_kind_eq_basic() {
    let nodes = vec![1u32, 2, 1, 3, 1, 4];
    let kind = 1u32;
    let cpu = kind_eq_cpu(&nodes, kind);
    let gpu = run_node_kind_eq(&nodes, kind);
    assert_eq!(gpu, cpu);
    // Bits 0, 2, 4 should be set.
    assert_eq!(gpu, vec![0b010101]);
}

#[test]
fn cuda_node_kind_eq_no_matches() {
    let nodes = vec![1u32, 2, 3, 4];
    let kind = 99u32;
    let cpu = kind_eq_cpu(&nodes, kind);
    let gpu = run_node_kind_eq(&nodes, kind);
    assert_eq!(gpu, cpu);
    assert_eq!(gpu, vec![0u32]);
}

#[test]
fn cuda_node_kind_eq_all_match() {
    let nodes = vec![5u32; 8];
    let cpu = kind_eq_cpu(&nodes, 5);
    let gpu = run_node_kind_eq(&nodes, 5);
    assert_eq!(gpu, cpu);
    assert_eq!(gpu, vec![0b1111_1111]);
}

fn run_literal_of(nodes: &[u32]) -> Vec<u32> {
    let n = nodes.len() as u32;
    let program = literal_of("nodes", "nodeset", n);
    with_live_backend("literal predicate", |backend| {
        cuda_u32_bitset_output(backend, &program, n, nodes, "literal_of")
    })
}

#[test]
fn cuda_literal_of_matches_cpu() {
    let nodes = vec![1u32, 2, 3, 4, 5, 6, 7, 8];
    let cpu = literal_cpu(&nodes);
    let gpu = run_literal_of(&nodes);
    assert_eq!(gpu, cpu);
}