remove previous fix

Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
Sage Moore 2025-07-09 20:36:08 +00:00
parent b53450e368
commit 29a5ac1d04

View File

@ -280,10 +280,4 @@ class DeepEPLLAll2AllManager(DeepEPAll2AllManagerBase):
buffer_kwargs = self._make_all2all_kwargs(**kwargs)
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)
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()
return [first_handle, second_handle]