[Bugfix] Optimize CpuGpuBuffer initialization (#25447)

Signed-off-by: Naman Lalit <nl2688@nyu.edu>
2025-09-27 08:07:36 +08:00
parent dc48ba0c75
commit 4e33a7ea85
1 changed files with 1 additions and 1 deletions
@@ -117,7 +117,7 @@ class CpuGpuBuffer:
                               dtype=dtype,
                               device="cpu",
                               pin_memory=pin_memory)
-        self.gpu = self.cpu.to(device)
+        self.gpu = torch.zeros_like(self.cpu, device=device)
        self.np: np.ndarray
        # To keep type hints simple (avoiding generics and subclasses), we
        # only conditionally create the numpy array attribute. This can cause