The nvmath.bindings module is experimental and potentially subject to future changes.

newton_schulz_buffer_size

nvmath.bindings.cusolverMp.newton_schulz_buffer_size(

intptr_t handle,

intptr_t ns_desc,

int64_t m,

int64_t n,

intptr_t d_x,

int64_t ix,

int64_t jx,

intptr_t desc_x,

int64_t number_of_newton_schulz_iterations,

intptr_t h_coeffs,

int compute_type,

) → tuple

See cusolverMpNewtonSchulz_bufferSize.