Skip to content

Commit 26936b3

Browse files
authored
Fix async cp recipe (#3603)
## Description set checkpoint_future properly in pinned mem tutorial
1 parent 387be4c commit 26936b3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recipes_source/distributed_async_checkpoint_recipe.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,7 @@ checkpoint requests users can take advantage of direct memory access to speed up
263263
if checkpoint_future is not None:
264264
# waits for checkpointing to finish, avoiding queuing more then one checkpoint request at a time
265265
checkpoint_future.result()
266-
dcp.async_save(state_dict, storage_writer=writer, checkpoint_id=f"{CHECKPOINT_DIR}_step{step}")
266+
checkpoint_future = dcp.async_save(state_dict, storage_writer=writer, checkpoint_id=f"{CHECKPOINT_DIR}_step{step}")
267267
268268
cleanup()
269269

0 commit comments

Comments
 (0)