issue/86 - 添加计算结束前的同步函数

0a5f4d71 · pengcheng888 · efc19d7c · 0a5f4d71
Commit 0a5f4d71 authored Nov 26, 2025 by pengcheng888
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

python/infinilm/generation/utils.py python/infinilm/generation/utils.py +3 -0

No files found.
--- a/python/infinilm/generation/utils.py
+++ b/python/infinilm/generation/utils.py
@@ -224,6 +224,9 @@ class GenerationMixin:
                    1.0,
                    out=out,
                )
+
+            infinicore.sync_stream()  # 计算结束前需要同步
+
            end_time = time.time()
            time_list.append((end_time - start_time) * 1000)