From 1130d31c843f1f3c9eb54919c94ef0179dde6a6b Mon Sep 17 00:00:00 2001 From: Saiteja Samudrala Date: Mon, 29 Sep 2025 17:24:03 -0700 Subject: [PATCH] fix async cp recipe pinned mem tutorial --- recipes_source/distributed_async_checkpoint_recipe.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/distributed_async_checkpoint_recipe.rst b/recipes_source/distributed_async_checkpoint_recipe.rst index 0e9add51483..e959883a25b 100644 --- a/recipes_source/distributed_async_checkpoint_recipe.rst +++ b/recipes_source/distributed_async_checkpoint_recipe.rst @@ -263,7 +263,7 @@ checkpoint requests users can take advantage of direct memory access to speed up if checkpoint_future is not None: # waits for checkpointing to finish, avoiding queuing more then one checkpoint request at a time checkpoint_future.result() - dcp.async_save(state_dict, storage_writer=writer, checkpoint_id=f"{CHECKPOINT_DIR}_step{step}") + checkpoint_future = dcp.async_save(state_dict, storage_writer=writer, checkpoint_id=f"{CHECKPOINT_DIR}_step{step}") cleanup()