CupyBackendMemoryUsage

class CupyBackendMemoryUsage(fast_shape, ft_shape, float_nbytes, complex_nbytes, integer_nbytes)

Bases: MemoryProfile

tme.backends.CupyBackend memory estimator.

Attributes

CupyBackendMemoryUsage.base_complex	Number of shared complex arrays
CupyBackendMemoryUsage.base_float	Number of shared real arrays
CupyBackendMemoryUsage.fork_complex	Number of complex arrays per fork
CupyBackendMemoryUsage.fork_float	Number of real arrays per fork

Methods

CupyBackendMemoryUsage.base_usage()	Return the base memory usage in bytes.
CupyBackendMemoryUsage.per_fork()	Return the memory usage per fork in bytes.