modify unit test for fpdt

microsoft · Nov 5, 2024 · f570213 · f570213
1 parent 02c2fbf
commit f570213
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/sequence/fpdt_layer.py b/deepspeed/sequence/fpdt_layer.py
@@ -434,7 +434,7 @@ def __init__(self, chunk: torch.Tensor, device=None, is_in_use=False):
         self.gpu_chunk = chunk if is_in_use else None
 
     def load_to_gpu(self):
-        assert self.gpu_chunk is None
+        # assert self.gpu_chunk is None
         if self.gpu_chunk is not None:
             pass
         else: