Sun Peng 860de0aa46 Feat/add runntime gpu test (#254) ... * feat: add gpu bench * feat/add allreduce runtime bench --------- Co-authored-by: sunpengsdu <sunpengsdu@gmail.com>		2023-09-01 13:38:01 +08:00
..
__init__.py	initial commit	2023-07-06 12:55:23 +08:00
checkpoint.py	initial commit	2023-07-06 12:55:23 +08:00
common.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
evaluation.py	fix(eval): StreamingDataset does not have an __len__ method. (#251)	2023-08-31 15:29:04 +08:00
gputest.py	Feat/add runntime gpu test (#254)	2023-09-01 13:38:01 +08:00
logger.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
megatron_timers.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
model_checkpoint.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
parallel.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
registry.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
simple_memory_profiler.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
storage_manager.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
timeout.py	[feat]: add pal reasoning script (#163)	2023-08-10 17:53:46 +08:00
writer.py	feat(utils/writer.py): support writer add_scalars for writing dict data (#257)	2023-09-01 13:24:46 +08:00