feature(nyz): add GPU utils (#788)

* feature(nyz): add basic mem profiler * polish(nyz): polish API comments * style(nyz): fix flake8 style issue(ci skip)
opendilab · Apr 8, 2024 · 1ed8afc · 1ed8afc
1 parent 7d05491
commit 1ed8afc
Show file tree

Hide file tree

Showing 4 changed files with 802 additions and 0 deletions.
diff --git a/ding/utils/__init__.py b/ding/utils/__init__.py
@@ -32,6 +32,7 @@
 from .fast_copy import fastcopy
 from .bfs_helper import get_vi_sequence
 from .normalizer_helper import DatasetNormalizer
+from .memory_helper import SimpleMemoryProfiler
 
 if ding.enable_linklink:  # False as default
     from .linklink_dist_helper import get_rank, get_world_size, dist_mode, dist_init, dist_finalize, \