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

Allow a keyboard interrupt to flush the queue

Browse Source
This commit is contained in:
Steven Robertson 2012-01-13 20:57:35 -05:00
parent 348525e036
commit 9a7b5db96e
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
worker.py
View File

@ -193,14 +193,18 @@ def run_jobs(r, rev, jobs):
retry.append(i)
return sidx, gpu_time, ftag, jpg
for sidx, job in enumerate(jobs):
while len(pending) > QUEUE_LENGTH:
yield pull(True)
ret = pull(False)
if ret:
yield ret
pending[sidx] = job
push(sidx, job)
try:
for sidx, job in enumerate(jobs):
while len(pending) > QUEUE_LENGTH:
yield pull(True)
ret = pull(False)
if ret:
yield ret
pending[sidx] = job
waiting.append(sidx)
push(sidx, job)
except KeyboardInterrupt:
print 'Interrupt received, flushing already-dispatched frames'
while pending:
yield pull(True)
@ -229,4 +233,3 @@ if __name__ == "__main__":
work('192.168.1.3')
else:
client(sys.argv[1], sys.argv[2:])