search

argonne-lcf / Megatron-DeepSpeed

Ongoing research training transformer language models at scale, including: BERT & GPT-2
Other
9 stars 12 forks source link

fixed dftracer compatibility #55

Open zhenghh04 opened 2 months ago

zhenghh04 commented 2 months ago

dlio_profiler has been changed to dftracer with new API changes. This fix is to update the API functions on Megatron-DeepSpeed accordingly