Convert .pyre_configuration.local to fast by default architecture] [b…

…atch:67/72] [shard:2/N] [A] Reviewed By: connernilsen Differential Revision: D63910426 fbshipit-source-id: f4b4e06b3aeef8038e28ad359c386bfb06bf34b2
pytorch · Oct 4, 2024 · 1ff027d · 1ff027d
1 parent 0440bdd
commit 1ff027d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/torchsnapshot/serialization.py b/torchsnapshot/serialization.py
@@ -226,7 +226,6 @@ def _tensor_as_memoryview_via_untyped_storage(tensor: torch.Tensor) -> memoryvie
     return memoryview(tensor.numpy()).cast("b")
 
 
-# pyre-ignore[11]
 def contiguous_view_as_untyped_storage(tensor: torch.Tensor) -> UntypedStorage:
     if not tensor.is_contiguous():
         raise AssertionError(

diff --git a/torchsnapshot/uvm_tensor.py b/torchsnapshot/uvm_tensor.py
@@ -13,6 +13,7 @@
 _UVM_TENSOR_AVAILABLE = False
 
 try:
+    # pyre-fixme[21]: Could not find module `fbgemm_gpu`.
     import fbgemm_gpu  # @manual  # noqa
 except Exception:
     pass