Skip to content

Commit 53e83f1

Browse files
authored
fix: wrong accelerator name when node_rank is not set (RLinf#370)
Signed-off-by: Hao Lin <linhaomails@gmail.com>
1 parent 0d17b90 commit 53e83f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

rlinf/scheduler/cluster/node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -428,7 +428,7 @@ def _sort_nodes(self, cluster_num_nodes: int):
428428
# NODE_RANK not set, sort first by accelerator type, then by IP
429429
nodes_group_by_accel: dict[str, list[NodeInfo]] = {}
430430
for node in self._nodes:
431-
accel_name = f"{node.accelerator_type.value}_{node.accelerator_model}"
431+
accel_name = node.accelerator_type
432432
nodes_group_by_accel.setdefault(accel_name, [])
433433
nodes_group_by_accel[accel_name].append(node)
434434
for accel_name in nodes_group_by_accel.keys():

0 commit comments

Comments
 (0)