Skip to content

Commit dd562d2

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent aedb8d1 commit dd562d2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

colossalai/checkpoint_io/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1142,7 +1142,7 @@ def gather_state_dict_fast(
11421142
ks.append(k)
11431143
ops.append(dist.P2POp(dist.irecv, buffer, dist.get_global_rank(group, i), group))
11441144
reqs = dist.batch_isend_irecv(ops)
1145-
for req in reqs: # len(reqs) maybe be different from len(ops) because of coalescing
1145+
for req in reqs: # len(reqs) maybe be different from len(ops) because of coalescing
11461146
req.wait()
11471147
for k in ks:
11481148
returned_state_dict[k] = returned_state_dict[k].to(device)

0 commit comments

Comments
 (0)