mircuda 0.1.0

Native, explicit-stream Rust gateway to NVIDIA CUDA
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

use mircuda::{KernelSource, cuda_kernel, cuda_kernel_file, cuda_kernel_files};

#[test]
fn embeds_inline_source() {
    let source = cuda_kernel!(r#"extern "C" __global__ void probe() {}"#);
    assert_eq!(source.name(), "inline.cu");
    assert!(source.source().contains("probe"));
}

#[test]
fn embeds_file_source() {
    const SOURCE: KernelSource = cuda_kernel_file!("../kernels/probe.cu");
    assert_eq!(SOURCE.name(), "../kernels/probe.cu");
    assert!(SOURCE.source().contains("mircuda_probe"));
}

#[test]
fn composes_embedded_fragments_in_order() {
    const SOURCE: KernelSource = cuda_kernel_files!(
        "composed.cu";
        "../kernels/prefix.cuh",
        "../kernels/probe.cu",
    );
    let source = SOURCE.source();
    assert!(matches!(
        (source.find("MIRCUDA_PREFIX"), source.find("mircuda_probe")),
        (Some(prefix), Some(probe)) if prefix < probe
    ));
}