This commit is contained in:
iperov 2021-10-01 19:24:22 +04:00
commit 03212982e6

View file

@ -210,6 +210,8 @@ class Device:
""" """
pool = self._pooled_buffers pool = self._pooled_buffers
mems = [ (k,x) for k in pool.keys() for x in pool[k] ] mems = [ (k,x) for k in pool.keys() for x in pool[k] ]
if len(mems) != 0:
mems = random.sample(mems, max(1,int(len(mems)*0.25)) ) mems = random.sample(mems, max(1,int(len(mems)*0.25)) )
for k, mem in mems: for k, mem in mems:
self._cl_mem_free(mem) self._cl_mem_free(mem)