suppress errors in torchsnapshot

Differential Revision: D49531510 fbshipit-source-id: ff8e0052107a36dbe5a4640c3d6fa251f8e37988
pytorch · Sep 22, 2023 · e5cfe05 · e5cfe05
1 parent cde8f18
commit e5cfe05
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchsnapshot/batcher.py b/torchsnapshot/batcher.py
@@ -140,6 +140,7 @@ def __init__(
 
     async def stage_buffer(self, executor: Optional[Executor] = None) -> BufferType:
         try:
+            # pyre-fixme[16]: Module `cuda` has no attribute `ByteTensor`.
             gpu_buf = torch.cuda.ByteTensor(self.slab_sz_bytes)
         except torch.cuda.OutOfMemoryError:
             gpu_buf = None