mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 03:30:05 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add right buffer. 2x allocation of uchar buffer.

Browse Source
This commit is contained in:
Steven Robertson 2017-05-02 00:11:03 -07:00
parent 3a3b3b33d1
commit 8f21ffd4c3
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
cuburn/render.py
View File

@ -101,14 +101,15 @@ class Framebuffers(object):
self.d_points = cuda.mem_alloc(self._len_d_points) self.d_points = cuda.mem_alloc(self._len_d_points)
def _clear(self): def _clear(self):
self.nbins = self.d_front = self.d_back = self.d_left = self.d_uchar = None self.nbins = self.d_front = self.d_back = None
self.d_left = self.d_right = self.d_uchar = None
def free(self, stream=None): def free(self, stream=None):
if stream is not None: if stream is not None:
stream.synchronize() stream.synchronize()
else: else:
cuda.Context.synchronize() cuda.Context.synchronize()
for p in (self.d_front, self.d_back, self.d_left, self.d_uchar): for p in (self.d_front, self.d_back, self.d_left, self.d_right, self.d_uchar):
if p is not None: if p is not None:
p.free() p.free()
self._clear() self._clear()
@ -128,7 +129,8 @@ class Framebuffers(object):
self.d_front = cuda.mem_alloc(16 * nbins) self.d_front = cuda.mem_alloc(16 * nbins)
self.d_back = cuda.mem_alloc(16 * nbins) self.d_back = cuda.mem_alloc(16 * nbins)
self.d_left = cuda.mem_alloc(16 * nbins) self.d_left = cuda.mem_alloc(16 * nbins)
self.d_uchar = cuda.mem_alloc(nbins) self.d_right = cuda.mem_alloc(16 * nbins)
self.d_uchar = cuda.mem_alloc(2 * nbins)
self.nbins = nbins self.nbins = nbins
except cuda.MemoryError, e: except cuda.MemoryError, e:
# If a frame that's too large sneaks by the task distributor, we # If a frame that's too large sneaks by the task distributor, we
@ -157,6 +159,10 @@ class Framebuffers(object):
"""Flip the front and back buffers.""" """Flip the front and back buffers."""
self.d_front, self.d_back = self.d_back, self.d_front self.d_front, self.d_back = self.d_back, self.d_front
def flip_side(self):
"""Flip the left and right buffers."""
self.d_left, self.d_right = self.d_right, self.d_left
class DevSrc(object): class DevSrc(object):
""" """
The buffers which represent a genome on-device, in the formats needed to The buffers which represent a genome on-device, in the formats needed to