Skip to main content

Module elementwise

Module elementwise 

Source
Expand description

ElementwiseBinaryRequest<T> and ElementwiseTrinaryRequest<T>.

Both wrap the cuTENSOR cutensorCreate{Binary,Trinary} + cutensorElementwise{Binary,Trinary}Execute pair via our local crate::sys::cutensor wrappers — cudarc’s safe layer doesn’t expose these.

Structs§

ElementwiseBinaryRequest
D = op_AC( alpha * op_A(A), gamma * op_C(C) ).
ElementwiseTrinaryRequest
D = op_ABC( op_AB(alpha * op_A(A), beta * op_B(B)), gamma * op_C(C) ).

Functions§

build_binary_key_raw
build_trinary_key_raw