#6644: sync before calling print l1 buffers

tenstorrent · Apr 1, 2024 · ebdb284 · ebdb284
1 parent 8f3038a
commit ebdb284
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/ttnn/unit_tests/test_reports.py b/tests/ttnn/unit_tests/test_reports.py
@@ -29,6 +29,8 @@ def test_print_l1_buffers_of_add_operation(tmp_path, height, width):
 
     ttnn.to_torch(output_tensor)
 
+    ttnn.synchronize_device(device)
+
     ttnn.print_l1_buffers(str(tmp_path / "l1_buffer_state.txt"))
 
     device = ttnn.close_device(device)