ReshapeOptions#

class nvmath.distributed.reshape.ReshapeOptions(
logger: Logger | None = None,
blocking: Literal[True, 'auto'] = 'auto',
)[source]#

A data class for providing options to the Reshape object and the wrapper function reshape().

logger#

Python Logger object. The root logger will be used if a logger object is not provided.

Type:

logging.Logger

blocking#

A flag specifying the behavior of the execution functions and methods, such as reshape() and Reshape.execute(). When blocking is True, the execution methods do not return until the operation is complete. When blocking is "auto", the methods return immediately when the input tensor is on the GPU. The execution methods always block when the input tensor is on the CPU to ensure that the user doesn’t inadvertently use the result before it becomes available. The default is "auto".

Type:

Literal[True, ‘auto’]

See also

Reshape and reshape().