Skip to content

GitLab

Unverified Commit 3c6ad521 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #87 from InfiniTensor/issue/86 

issue/86 - 添加计算结束前的同步函数
parents efc19d7c 0a5f4d71
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
python/infinilm/generation/utils.py
View file @ 3c6ad521
...@@ -224,6 +224,9 @@ class GenerationMixin: ...@@ -224,6 +224,9 @@ class GenerationMixin:
1.0, 1.0,
out=out, out=out,
) )
infinicore.sync_stream() # 计算结束前需要同步
end_time = time.time() end_time = time.time()
time_list.append((end_time - start_time) * 1000) time_list.append((end_time - start_time) * 1000)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment