Add missing Pyre mode headers] [batch:28/1458] [shard:22/N]

Differential Revision: D66867205

fbshipit-source-id: 872f92d84eb8561a95613d49638f0f6b499285ff

tb_plugin/torch_tb_profiler/__init__.py

@@ -2,6 +2,8 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
 
+# pyre-unsafe
+
 # Entry point for Pytorch TensorBoard plugin package.
 
 __version__ = '0.4.3'

tb_plugin/torch_tb_profiler/consts.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 import re
 from collections import namedtuple
 

tb_plugin/torch_tb_profiler/io/__init__.py

@@ -1,3 +1,4 @@
+# pyre-unsafe
 from .cache import Cache
 from .file import (BaseFileSystem, StatData, abspath, basename, download_file,
                    exists, get_filesystem, glob, isdir, join, listdir,

tb_plugin/torch_tb_profiler/io/azureblob.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import os
 
 # pyre-fixme[21]: Could not find module `azure.storage.blob`.

tb_plugin/torch_tb_profiler/io/base.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import os
 from abc import ABC, abstractmethod
 from collections import namedtuple

tb_plugin/torch_tb_profiler/io/cache.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import tempfile
 
 from .. import utils

tb_plugin/torch_tb_profiler/io/file.py

@@ -1,3 +1,4 @@
+# pyre-unsafe
 """
 This file is forked from
 https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/tensorflow_stub/io/gfile.py.

tb_plugin/torch_tb_profiler/io/gs.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 # pyre-fixme[21]: Could not find module `google.cloud`.
 from google.cloud import storage
 # pyre-fixme[21]: Could not find module `google.auth`.

tb_plugin/torch_tb_profiler/io/hdfs.py

@@ -1,3 +1,4 @@
+# pyre-unsafe
 import os
 
 # pyre-fixme[21]: Could not find module `fsspec`.

tb_plugin/torch_tb_profiler/io/utils.py

@@ -1,3 +1,4 @@
+# pyre-unsafe
 def as_str_any(value):
     """Converts to `str` as `str(value)`, but use `as_str` for `bytes`.
 

tb_plugin/torch_tb_profiler/multiprocessing.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import multiprocessing as mp
 import os
 

tb_plugin/torch_tb_profiler/plugin.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 import atexit
 import gzip
 import json

tb_plugin/torch_tb_profiler/profiler/__init__.py

@@ -2,6 +2,8 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
 
+# pyre-unsafe
+
 from .loader import RunLoader
 
 __all__ = ['RunLoader']

tb_plugin/torch_tb_profiler/profiler/communication.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 from typing import Dict, List, Tuple
 
 from .. import utils

tb_plugin/torch_tb_profiler/profiler/data.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 import gzip
 import io as sysio
 import json

tb_plugin/torch_tb_profiler/profiler/diffrun/__init__.py

@@ -1,3 +1,4 @@
+# pyre-unsafe
 from .contract import DiffStats, OpAgg
 from .tree import (DiffNode, compare_op_tree, diff_summary, print_node,
                    print_ops)

tb_plugin/torch_tb_profiler/profiler/diffrun/contract.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 from collections import namedtuple
 from typing import Dict, List
 

tb_plugin/torch_tb_profiler/profiler/diffrun/operator.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 from abc import ABCMeta
 from typing import List, Tuple, Union
 

tb_plugin/torch_tb_profiler/profiler/diffrun/tree.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import sys
 from typing import Generator, List, Union
 

tb_plugin/torch_tb_profiler/profiler/event_parser.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import sys
 from collections import defaultdict
 from enum import IntEnum

tb_plugin/torch_tb_profiler/profiler/gpu_metrics_parser.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from typing import Iterable, List
 
 from .. import consts, utils

tb_plugin/torch_tb_profiler/profiler/kernel_parser.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from typing import Optional
 
 import numpy as np

tb_plugin/torch_tb_profiler/profiler/loader.py

@@ -1,3 +1,4 @@
+# pyre-unsafe
 
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.

tb_plugin/torch_tb_profiler/profiler/memory_parser.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from collections import defaultdict
 from enum import IntEnum
 from typing import Dict, Iterable, List, Optional, Tuple

tb_plugin/torch_tb_profiler/profiler/module_op.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 from collections import namedtuple
 from typing import Dict, Generator, Iterable, List, Optional, Set, Tuple, Union
 

tb_plugin/torch_tb_profiler/profiler/node.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import sys
 from abc import ABC
 from typing import List, Optional, Tuple

tb_plugin/torch_tb_profiler/profiler/op_agg.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 import sys
 from collections import defaultdict
 from typing import Callable, Dict, List

tb_plugin/torch_tb_profiler/profiler/op_tree.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 import sys
 from collections import defaultdict
 from typing import Dict, Iterable, List, Optional, Tuple

tb_plugin/torch_tb_profiler/profiler/overall_parser.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from typing import List, Tuple
 
 from .. import utils

tb_plugin/torch_tb_profiler/profiler/range_utils.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 from typing import List, Tuple
 
 

tb_plugin/torch_tb_profiler/profiler/run_generator.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from collections import OrderedDict
 from typing import Dict, Iterable, List
 

tb_plugin/torch_tb_profiler/profiler/tensor_core.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 class TC_Allowlist_Meta(type):
     # Enable grammar sugar as 'v in TC_Allowlist'.
     def __contains__(cls, item):

tb_plugin/torch_tb_profiler/profiler/tensor_cores_parser.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # -------------------------------------------------------------------------
+
+# pyre-unsafe
 from typing import Dict, Iterable, List
 
 from .. import consts

tb_plugin/torch_tb_profiler/profiler/trace.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from enum import IntEnum
 from typing import Dict, Optional
 

tb_plugin/torch_tb_profiler/run.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 from collections import defaultdict
 from typing import Any, Dict, Iterable, List, Optional, Tuple, Union
 

tb_plugin/torch_tb_profiler/utils.py

@@ -1,6 +1,8 @@
 # -------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # --------------------------------------------------------------------------
+
+# pyre-unsafe
 import logging
 import math
 import os