Revert "Arm backend: Remove pyre-unsafe from tosa/, vgf/ and ethosu/ (pytorch#15352)"

This reverts commit eff4024.

Author: GregoryComer

backends/arm/ethosu/__init__.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 #
+# pyre-unsafe
 
 from .backend import EthosUBackend  # noqa: F401
 from .compile_spec import EthosUCompileSpec  # noqa: F401

backends/arm/ethosu/backend.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 
 #
 # Main implementation of AoT flow to partition and preprocess for Arm target

backends/arm/ethosu/partitioner.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 
 from typing import final, Optional, Sequence
 

backends/arm/tosa/__init__.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 #
+# pyre-unsafe
 
 from .specification import TosaSpecification
 

backends/arm/tosa/backend.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 
 #
 # Main implementation of AoT flow to partition and preprocess for Arm target

backends/arm/tosa/mapping.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 """Provide PyTorch-to-TOSA mapping helpers.
 
 Use these utilities to translate PyTorch dtypes and FX node metadata into

backends/arm/tosa/partitioner.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 """Provide a partitioner for delegating subgraphs to the TOSA backend.
 
 Implement logic to identify and tag regions of an ``ExportedProgram`` that can

backends/arm/tosa/quant_utils.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 
 # Utility functions for TOSA quantized lowerings
 

backends/arm/tosa/specification.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 """Provide TOSA specification parsing and context utilities.
 
 Use these helpers to parse and validate TOSA profile/extension strings and to

backends/arm/tosa/utils.py

@@ -3,6 +3,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-unsafe
 
 import logging
 from typing import Any