Added queue-finish commands to PyCLBlast samples and tests

src/pyclblast/samples/saxpy.py

@@ -32,5 +32,6 @@ cly.set(y)
 
 print("# Example level-1 operation: AXPY")
 pyclblast.axpy(queue, n, clx, cly, alpha=alpha)
+queue.finish()
 print("# Result for vector y: %s" % cly.get())
 print("# Expected result:     %s" % (alpha * x + y))

src/pyclblast/samples/sgemm.py

@@ -34,5 +34,6 @@ clc.set(c)
 
 print("# Example level-3 operation: GEMM")
 pyclblast.gemm(queue, m, n, k, cla, clb, clc, a_ld=k, b_ld=n, c_ld=n)
+queue.finish()
 print("# Matrix C result: %s" % clc.get())
 print("# Expected result: %s" % (np.dot(a, b)))

src/pyclblast/samples/sgemv.py

@@ -36,5 +36,6 @@ cly.set(y)
 
 print("# Example level-2 operation: GEMV")
 pyclblast.gemv(queue, m, n, cla, clx, cly, a_ld=n, alpha=alpha, beta=beta)
+queue.finish()
 print("# Result for vector y: %s" % cly.get())
 print("# Expected result:     %s" % (alpha * np.dot(a, x) + beta * y))

src/pyclblast/test/test_pyclblast.py

@@ -32,6 +32,7 @@ class TestPyCLBlast(unittest.TestCase):
             opencl_array.set(numpy_array)
             host_arrays.append(numpy_array)
             device_arrays.append(opencl_array)
+        queue.finish()
         return queue, host_arrays, device_arrays
 
     def test_axpy(self):
@@ -40,6 +41,7 @@ class TestPyCLBlast(unittest.TestCase):
                 for n in [1, 7, 32]:
                     queue, h, d = self.setup([(n,), (n,)], dtype=dtype)
                     pyclblast.axpy(queue, n, d[0], d[1], alpha=alpha)
+                    queue.finish()
                     result = d[1].get()
                     reference = alpha * h[0] + h[1]
                     for i in range(n):
@@ -54,6 +56,7 @@ class TestPyCLBlast(unittest.TestCase):
                             queue, h, d = self.setup([(m, n), (n,), (m,)], dtype=dtype)
                             pyclblast.gemv(queue, m, n, d[0], d[1], d[2],
                                            a_ld=n, alpha=alpha, beta=beta)
+                            queue.finish()
                             result = d[2].get()
                             reference = alpha * np.dot(h[0], h[1]) + beta * h[2]
                             for i in range(m):
@@ -69,6 +72,7 @@ class TestPyCLBlast(unittest.TestCase):
                                 queue, h, d = self.setup([(m, k), (k, n), (m, n)], dtype=dtype)
                                 pyclblast.gemm(queue, m, n, k, d[0], d[1], d[2],
                                                a_ld=k, b_ld=n, c_ld=n, alpha=alpha, beta=beta)
+                                queue.finish()
                                 result = d[2].get()
                                 reference = alpha * np.dot(h[0], h[1]) + beta * h[2]
                                 for i in range(m):