Add missing Pyre mode headers] [batch:8/1557] [shard:11/N]

Differential Revision: D69040464 fbshipit-source-id: 9fe8f48ef45fe258617495935520139152947bcc
pytorch · Feb 4, 2025 · f18ec58 · f18ec58
1 parent 01d4c68
commit f18ec58
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/fbgemm_gpu/fbgemm_gpu/tbe_input_multiplexer.py b/fbgemm_gpu/fbgemm_gpu/tbe_input_multiplexer.py
@@ -5,6 +5,8 @@
 # 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 abc
 
 from dataclasses import dataclass

diff --git a/fbgemm_gpu/test/tbe/utils/generate_vbe_metadata_test.py b/fbgemm_gpu/test/tbe/utils/generate_vbe_metadata_test.py
@@ -5,6 +5,8 @@
 # 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 unittest
 from itertools import accumulate