Pyre Faster Type Checking By Default Mode Headers] [batch:1/188] [sha…

…rd:3/N] (pytorch#2704) Summary: Pull Request resolved: pytorch#2704 Reviewed By: connernilsen Differential Revision: D58348553 fbshipit-source-id: 4f16abdcee6e6f19cc328973cbc91398a8d51d6b
q10 · Jun 10, 2024 · 98b7780 · 98b7780
1 parent c7eea48
commit 98b7780
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/fbgemm_gpu/fbgemm_gpu/tbe/cache/__init__.py b/fbgemm_gpu/fbgemm_gpu/tbe/cache/__init__.py
@@ -5,4 +5,6 @@
 # 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 .split_embeddings_cache_ops import get_unique_indices_v2  # noqa: F401
diff --git a/fbgemm_gpu/fbgemm_gpu/tbe/cache/split_embeddings_cache_ops.py b/fbgemm_gpu/fbgemm_gpu/tbe/cache/split_embeddings_cache_ops.py
@@ -4,6 +4,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
+
 from typing import Optional, Tuple, Union
 
 import torch