Unverified Commit 7f57769c authored by Atream's avatar Atream Committed by GitHub
Browse files

Merge pull request #852 from Lander-Hatsune/main

cpuinfer: filter repeated backend instantiation
parents 1c001b80 d166fb9f
...@@ -727,8 +727,13 @@ class CPUInferKVCache: ...@@ -727,8 +727,13 @@ class CPUInferKVCache:
class CPUInfer: class CPUInfer:
cpuinfer = None cpuinfer = None
cur_backend_thread_num = 0
def __init__(self, thread_num): def __init__(self, thread_num):
CPUInfer.cpuinfer = cpuinfer_ext.CPUInfer(thread_num) if thread_num > CPUInfer.cur_backend_thread_num:
CPUInfer.cur_backend_thread_num = thread_num
del CPUInfer.cpuinfer
CPUInfer.cpuinfer = cpuinfer_ext.CPUInfer(thread_num)
def submit(self, task): def submit(self, task):
CPUInfer.cpuinfer.submit(task) CPUInfer.cpuinfer.submit(task)
......
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