Skip to content

Commit 1ed8afc

Browse files
authored
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)
1 parent 7d05491 commit 1ed8afc

File tree

4 files changed

+802
-0
lines changed

4 files changed

+802
-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)