• Hongtao Zhang's avatar
    Enhancement: Add nsys and pytorch profiler debug trace support (#744) · d804dbb6
    Hongtao Zhang authored
    
    
    To improve benchmark debugging, the following debug methods were added:
    
    pytorch profiler in model benchmark
    
    - SB_ENABLE_PYTORCH_PROFILER: switch to enable/disable
    - SB_TORCH_PROFILER_TRACE_DIR: log path
    These 2 runtime variables need to be configured in SB config file.
    
    nsys in SB runner
    
    - SB_ENABLE_NSYS: switch to enable/disable 
    - SB_NSYS_TRACE_DIR: log path
    These 2 runtime variables need to be configured in runner's ENV
    
    ---------
    Co-authored-by: default avatarHongtao Zhang <hongtaozhang@microsoft.com>
    d804dbb6
pytorch_base.py 15.5 KB