Skip to content

Commit

Permalink
update python style
Browse files Browse the repository at this point in the history
  • Loading branch information
songhappy committed Nov 27, 2024
1 parent 2b8c289 commit 4eb7fe5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion torchtune/utils/_device.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,8 @@ class DeviceSupport(Enum):
The following enumeration defines various device configurations with attributes:
1. `device_type` (str): The type of device (e.g., "cpu", "cuda", "npu", "xpu").
2. `device_name` (str): A user-friendly name for the device (e.g., "CPU", "GPU", "NPU", "XPU").
3. `communication_backend` (str): Specifies the backend used for communication on this device (e.g., "gloo", "nccl", "hccl", "ccl").
3. `communication_backend` (str): Specifies the backend used for communication on this device
(e.g., "gloo", "nccl", "hccl", "ccl").
"""

CPU = ("cpu", "CPU", "gloo")
Expand Down

0 comments on commit 4eb7fe5

Please sign in to comment.