Skip to main content

reduce

Function reduce 

Source
pub unsafe fn reduce(
    handle: cutensorHandle_t,
    plan: cutensorPlan_t,
    alpha: *const c_void,
    a: *const c_void,
    beta: *const c_void,
    c: *const c_void,
    d: *mut c_void,
    workspace: *mut c_void,
    workspace_size: u64,
    stream: cudaStream_t,
) -> Result<(), CutensorError>
Expand description

Wraps cutensorReduce (post-plan execution).

ยงSafety

All pointers must be valid. workspace must hold at least workspace_size bytes. stream must outlive the call.