Fix: add event sync in fsdp tutorial (#3166)

zzb610 · svekars · web-flow · commit 33a52a5191dd · 2025-01-08T08:37:41.000-08:00
Signed-off-by: zzb610 &lt;bwyuchi@gmail.com&gt;
Co-authored-by: Svetlana Karslioglu &lt;svekars@meta.com&gt;
diff --git a/intermediate_source/FSDP_tutorial.rst b/intermediate_source/FSDP_tutorial.rst
@@ -251,6 +251,7 @@ We add the following code snippets to a python script “FSDP_mnist.py”.
         init_end_event.record()
 
         if rank == 0:
+            init_end_event.synchronize()
             print(f"CUDA event elapsed time: {init_start_event.elapsed_time(init_end_event) / 1000}sec")
             print(f"{model}")
 
