• Salil Desai's avatar
    Clean Up MobileOptimizerType Rewrite Flags Public API and Documentation · 18de6ffb
    Salil Desai authored
    Summary:
    X-link: https://github.com/pytorch/pytorch/pull/91600
    
    Pull Request resolved: https://github.com/facebookresearch/d2go/pull/452
    
    bypass-github-export-checks
    
    Remove MobileOptimizerType and all rewrite flags from torch.X and torch._C.X to clean up torch.X and torch._C.X namespaces
    
    The affected rewrite flags are
    - CONV_BN_FUSION
    - FUSE_ADD_RELU
    - HOIST_CONV_PACKED_PARAMS
    - INSERT_FOLD_PREPACK_OPS
    - REMOVE_DROPOUT
    - VULKAN_AUTOMATIC_GPU_TRANSFER
    
    Bc-Breaking Change:
    
    Before this change, the rewrite flags were accessible through all of
    1. torch.utils.mobile_optimizer.MobileOptimizerType.X
    2. torch._C.MobileOptimizerType.X
    3. torch.X
    4. torch.MobileOptimizerType.X
    5. torch._C.X
    
    But after this change, only torch.utils.mobile_optimizer.MobileOptimizerType.X  (option 1 above) and the newly added torch._C._MobileOptimizerType.X remain
    
    Corresponding updates to PyTorch Tutorial Docs are in https://github.com/pytorch/tutorials/pull/2163
    
    Reviewed By: kimishpatel
    
    Differential Revision: D41690203
    
    fbshipit-source-id: d901bdcbd16a594c3268e09b57c61b38c33a562f
    18de6ffb
torchscript.py 18.8 KB