From 03212982e6f0343aca01abef49a26742225aa093 Mon Sep 17 00:00:00 2001 From: iperov Date: Fri, 1 Oct 2021 19:24:22 +0400 Subject: [PATCH] fix --- xlib/avecl/_internal/backend/Device.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/xlib/avecl/_internal/backend/Device.py b/xlib/avecl/_internal/backend/Device.py index bb5dbc4..7043d0b 100644 --- a/xlib/avecl/_internal/backend/Device.py +++ b/xlib/avecl/_internal/backend/Device.py @@ -210,10 +210,12 @@ class Device: """ pool = self._pooled_buffers mems = [ (k,x) for k in pool.keys() for x in pool[k] ] - mems = random.sample(mems, max(1,int(len(mems)*0.25)) ) - for k, mem in mems: - self._cl_mem_free(mem) - pool[k].remove(mem) + + if len(mems) != 0: + mems = random.sample(mems, max(1,int(len(mems)*0.25)) ) + for k, mem in mems: + self._cl_mem_free(mem) + pool[k].remove(mem) return len(mems) != 0 def _keep_target_memory_usage(self):