Skip to content

memory_utils#

Includes utility functions for memory operations.

Functions:

print_memory_usage #

print_memory_usage() -> None

Print CPU/GPU memory usage for rank 0.