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

newton_schulz

nvmath.bindings.cusolverMp.newton_schulz(

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,

intptr_t d_work,

size_t workspace_in_bytes_on_device,

intptr_t h_work,

size_t workspace_in_bytes_on_host,

intptr_t info,

)

See cusolverMpNewtonSchulz.