pyre-fixmes for D75477355 - torchsnapshot

Summary: Additional pyre-fixmes needed by pytorch change D75477355 that weren't caught by diff-time CI.

Reviewed By: bobrenjc93

Differential Revision: D77120303

fbshipit-source-id: 940d4bb0e06f6d3b6eba10f7621fa578d2db543a

Author: aorenste

torchsnapshot/uvm_tensor.py

@@ -25,8 +25,14 @@
 
 
 try:
+    # pyre-fixme[9]: new_managed_tensor has type `(t: Tensor, sizes: List[int]) ->
+    #  Tensor`; used as `OpOverloadPacket`.
     new_managed_tensor = torch.ops.fbgemm.new_managed_tensor
+    # pyre-fixme[9]: is_uvm_tensor has type `(t: Tensor) -> bool`; used as
+    #  `OpOverloadPacket`.
     is_uvm_tensor = torch.ops.fbgemm.is_uvm_tensor
+    # pyre-fixme[9]: uvm_to_cpu has type `(t: Tensor) -> Tensor`; used as
+    #  `OpOverloadPacket`.
     uvm_to_cpu = torch.ops.fbgemm.uvm_to_cpu
 
     _UVM_TENSOR_AVAILABLE = True