Skip to main content

cubecl_std/throughput/runners/
launch_overhead.rs

1use cubecl::prelude::*;
2use cubecl_core as cubecl;
3
4#[cube(launch_unchecked)]
5pub fn launch_overhead<I: Numeric, N: Size>(
6    input: &[Vector<I, N>],
7    output: &mut [Vector<I, N>],
8    #[define(I)] _dtype: StorageType,
9) {
10    if ABSOLUTE_POS == 0 {
11        output[0] = input[0];
12    }
13}