Method set_device

colossalai/accelerator/cpu_accelerator.py:45–49

Bind the current process to a device.

(self, device: Optional[Union[torch.device, int]] = None)

        raise RuntimeError("this method is not supported for cpu accelerator")

    def set_device(self, device: Optional[Union[torch.device, int]] = None) -> None:
        """
        Bind the current process to a device.
        """
        raise RuntimeError("this method is not supported for cpu accelerator")

    def get_device_name(self, device: Union[torch.device, int]) -> str:
        """