CupyBackendMemoryUsage

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

Bases: MatchingMemoryUsage

Memory usage estimation for CupyBackend.

See also

tme.backends.CupyBackend

Methods

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