update

hpcaitech · Dec 11, 2023 · 4dc4852 · 4dc4852
1 parent 64b214f
commit 4dc4852
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/colossalai/accelerator/cuda_accelerator.py b/colossalai/accelerator/cuda_accelerator.py
@@ -39,7 +39,7 @@ def set_device(self, device: Union[torch.device, int]) -> None:
         """
         if device is None:
             if not dist.is_initialized():
-                raise RuntimeError("Cannot get current device when distributed is not initialized")
+                raise RuntimeError("Cannot get current device when distributed is not initialized.")
             device = dist.get_rank() % self.device_count()
         torch.cuda.set_device(device)
 

diff --git a/colossalai/accelerator/npu_accelerator.py b/colossalai/accelerator/npu_accelerator.py
@@ -48,7 +48,7 @@ def set_device(self, device: Union[torch.device, int]) -> None:
         """
         if device is None:
             if not dist.is_initialized():
-                raise RuntimeError("Cannot get current device when distributed is not initialized")
+                raise RuntimeError("Cannot get current device when distributed is not initialized.")
             device = dist.get_rank() % self.device_count()
         torch.npu.set_device(device)