Skip to content
Snippets Groups Projects
  • Qian Zhao's avatar
    0d25625b
    [Feature] Support torch.compile since PyTorch2.0 (#976) · 0d25625b
    Qian Zhao authored
    * enable compile configurations to support torch.compile in Runner
    
    * enable compilation in train, val and test
    
    * fix as comments
    
    * add docstring to illustrate usage
    
    * minor refine error message
    
    * add unittests
    
    * fix ut skip
    
    * add logging message to inform users
    
    * compile `train_step`, `val_step`, `test_step` instead
    
    * fix as comments
    
    * revert to compile `train_step` only due to pt2 issue
    
    * add documentation about torch.compile
    0d25625b
    History
    [Feature] Support torch.compile since PyTorch2.0 (#976)
    Qian Zhao authored
    * enable compile configurations to support torch.compile in Runner
    
    * enable compilation in train, val and test
    
    * fix as comments
    
    * add docstring to illustrate usage
    
    * minor refine error message
    
    * add unittests
    
    * fix ut skip
    
    * add logging message to inform users
    
    * compile `train_step`, `val_step`, `test_step` instead
    
    * fix as comments
    
    * revert to compile `train_step` only due to pt2 issue
    
    * add documentation about torch.compile