Apply ufmt to torch internal (pytorch#81643)

This is a big bang PR, merge conflicts are probably expected and will be addressed at merge.
Pull Request resolved: pytorch#81643
Approved by: https://github.com/ezyang

Author: huydhn

.lintrunner.toml

@@ -689,6 +689,13 @@ code = 'UFMT'
 include_patterns = [
     'test/onnx/**/*.py',
     'tools/**/*.py',
+    'torch/_decomp/**/*.py',
+    'torch/_lazy/**/*.py',
+    'torch/_masked/**/*.py',
+    'torch/_prims/**/*.py',
+    'torch/_refs/**/*.py',
+    'torch/_subclasses/**/*.py',
+    'torch/_*.py',
     'torchgen/**/*.py',
 ]
 command = [
@@ -699,6 +706,7 @@ command = [
 ]
 exclude_patterns = [
     'tools/gen_vulkan_spv.py',
+    'torch/__init__.py',  # Skip this file to format because it's part of the public API
 ]
 init_command = [
     'python3',

torch/_VF.py

@@ -10,10 +10,11 @@
 introducing torch._VF
 
 """
-import torch
 import sys
 import types
 
+import torch
+
 
 class VFModule(types.ModuleType):
     vf: types.ModuleType

torch/__config__.py

@@ -8,13 +8,15 @@ def show():
     """
     return torch._C._show_config()
 
+
 # TODO: In principle, we could provide more structured version/config
 # information here. For now only CXX_FLAGS is exposed, as Timer
 # uses them.
 def _cxx_flags():
     """Returns the CXX_FLAGS used when building PyTorch."""
     return torch._C._cxx_flags()
 
+
 def parallel_info():
     r"""Returns detailed string with parallelization settings"""
     return torch._C._parallel_info()

torch/__future__.py

@@ -11,9 +11,11 @@
 """
 _overwrite_module_params_on_conversion = False
 
+
 def set_overwrite_module_params_on_conversion(value):
     global _overwrite_module_params_on_conversion
     _overwrite_module_params_on_conversion = value
 
+
 def get_overwrite_module_params_on_conversion():
     return _overwrite_module_params_on_conversion