Skip to content

Commit 8a5a998

Browse files
committed
feature(nyz): add basic mem profiler
1 parent 7d05491 commit 8a5a998

File tree

4 files changed

+763
-0
lines changed

4 files changed

+763
-0
lines changed

ding/utils/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
from .fast_copy import fastcopy
3333
from .bfs_helper import get_vi_sequence
3434
from .normalizer_helper import DatasetNormalizer
35+
from .memory_helper import SimpleMemoryProfiler
3536

3637
if ding.enable_linklink: # False as default
3738
from .linklink_dist_helper import get_rank, get_world_size, dist_mode, dist_init, dist_finalize, \

0 commit comments

Comments
 (0)