mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-28 06:57:09 +08:00
remove previous fix
Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
parent
b53450e368
commit
29a5ac1d04
@ -280,10 +280,4 @@ class DeepEPLLAll2AllManager(DeepEPAll2AllManagerBase):
|
|||||||
buffer_kwargs = self._make_all2all_kwargs(**kwargs)
|
buffer_kwargs = self._make_all2all_kwargs(**kwargs)
|
||||||
first_handle = self.handle_caches[0].get_or_create(buffer_kwargs, deep_ep.Buffer)
|
first_handle = self.handle_caches[0].get_or_create(buffer_kwargs, deep_ep.Buffer)
|
||||||
second_handle = self.handle_caches[1].get_or_create(buffer_kwargs, deep_ep.Buffer)
|
second_handle = self.handle_caches[1].get_or_create(buffer_kwargs, deep_ep.Buffer)
|
||||||
return [first_handle, second_handle]
|
return [first_handle, second_handle]
|
||||||
|
|
||||||
def destroy(self):
|
|
||||||
for handle_cache in self.handle_caches:
|
|
||||||
with handle_cache._lock:
|
|
||||||
for _, handle in handle_cache._cache.items():
|
|
||||||
handle.destroy()
|
|
||||||
Loading…
x
Reference in New Issue
Block a user