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

get_newton_schulz_descriptor_attribute_dtype

nvmath.bindings.cusolverMp.get_newton_schulz_descriptor_attribute_dtype(int attr)

Get the Python data type of the corresponding NewtonSchulzDescriptorAttribute attribute.

Parameters:

attr (NewtonSchulzDescriptorAttribute) – The attribute to query.

Returns:

The data type of the queried attribute.

Note

This API has no C counterpart and is a convenient helper for allocating memory for newton_schulz_descriptor_get_attribute(), newton_schulz_descriptor_set_attribute().