Skip to main content

prepare

ruprim::reduce::launch::reduce_kernel

Function prepare 

Source 
pub fn prepare<'kernel, RA: ReduceArgs, __R: Runtime>(
    __client: &ComputeClient<__R>,
    __ruda_count: RudaCount,
    __ruda_dim: RudaDim,
    __address_type: AddressType,
    _in_size: usize,
    _out_size: usize,
    input: RuntimeArg<RA::Input<In, InSize>, __R>,
    output: RuntimeArg<RA::Output<Out, OutSize>, __R>,
    reduce_axis: RuntimeArg<usize, __R>,
    out_vec_axis: RuntimeArg<usize, __R>,
    blueprint: ReduceBlueprint,
    config: ReduceOperationConfig,
    _input_dtype: StorageType,
    _output_dtype: StorageType,
    _acc_dtype: StorageType,
) -> PreparedKernel<__R>
Expand description

Register owned arguments for explicit graph construction. No kernel is compiled or executed by this function.