From c59829ad86eadd1dd79acd51949c76402ccb17c1 Mon Sep 17 00:00:00 2001
From: Steven Robertson <steven@strobe.cc>
Date: Sat, 10 Dec 2011 16:23:24 -0500
Subject: [PATCH] Fix buffer overflow

---
 cuburn/code/iter.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cuburn/code/iter.py b/cuburn/code/iter.py
index c4220b8..30e0e02 100644
--- a/cuburn/code/iter.py
+++ b/cuburn/code/iter.py
@@ -435,6 +435,7 @@ oflow_end:
 __global__
 void flush_atom(uint64_t out_ptr, uint64_t atom_ptr, int nbins) {
     int i = (blockIdx.y * gridDim.x + blockIdx.x) * blockDim.x + threadIdx.x;
+    if (i >= nbins) return;
     asm volatile ({{crep("""
 {
     .reg .u32   off, hi, lo, d, r, g, b;