Skip to main content

Module launch

Module launch 

Source
Expand description

Kernel launch builder — the Rust equivalent of CUDA C’s triple-chevron syntax.

unsafe {
    kernel.launch()
        .grid((n.div_ceil(256), 1, 1))
        .block((256, 1, 1))
        .stream(&stream)
        .arg(&d_a.as_raw())
        .arg(&d_b.as_raw())
        .arg(&d_c.as_raw())
        .arg(&n)
        .launch()?;
}

Structs§

Dim3
Three-dimensional grid/block size.
LaunchBuilder
Builder produced by Function::launch. Call LaunchBuilder::launch to actually enqueue the kernel.