ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

Published 20 hours ago • verified publisher icon hpcaitech

[FEATURE]: integrate tracer

Open YuliangLiu0306 opened this issue 2 years ago • 0 comments

Overview

There are three version implementations of tracer. They solve different problems and fail in different scenarios. To make it clear, we need to integrate them into one framework. The integrated tracer should be able to handle different scenarios and provide a unified interface for users.

evaluation

The integrated tracer will be tested on following model zoos:

  • [ ] transformers #3157
  • [ ] torchrec
  • [ ] torchvision
  • [ ] timm
  • [ ] torchaudio

YuliangLiu0306 avatar Mar 17 '23 07:03 YuliangLiu0306