• ver217's avatar
    [utils] refactor profiler (#837) · 232142f4
    ver217 authored
    * add model data profiler
    
    * add a subclass of torch.profiler.profile
    
    * refactor folder structure
    
    * remove redundant codes
    
    * polish code
    
    * use GeminiMemoryManager
    
    * fix import path
    
    * fix stm profiler ext
    
    * polish comments
    
    * remove useless file
    232142f4
profiler.py 8.41 KB