From 4a603cfbd172ea71d37f5024450e11e4aaf94ae6 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 8 Sep 2009 01:26:05 -0400
Subject: [PATCH] Add (fixed) narray example by Roger Pau Monn'e.

---
 examples/narray.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 examples/narray.py

diff --git a/examples/narray.py b/examples/narray.py
new file mode 100644
index 00000000..c90f6cab
--- /dev/null
+++ b/examples/narray.py
@@ -0,0 +1,37 @@
+# example by Roger Pau Monn'e
+import pyopencl as cl
+import numpy as np
+
+demo_r = np.empty( (500,5), dtype=np.uint32)
+ctx = cl.create_context_from_type(cl.device_type.GPU)
+queue = cl.CommandQueue(ctx)
+
+mf = cl.mem_flags
+demo_buf = cl.create_buffer(ctx, mf.WRITE_ONLY, demo_r.nbytes)
+
+
+prg = cl.create_program_with_source(ctx,
+"""
+__kernel void demo(__global uint *demo)
+{
+    int i;
+    int gid = get_global_id(0);
+    for(i=0; i<5;i++)
+    {
+        demo[gid*5+i] = (uint) 1;
+    }
+}""")
+
+try:
+    prg.build()
+except:
+    print "Error:"
+    print prg.get_build_info(ctx.devices[0], cl.program_build_info.LOG)
+    raise
+
+prg.demo(queue, (500,), demo_buf)
+cl.enqueue_read_buffer(queue, demo_buf, demo_r).wait()
+
+for res in demo_r:
+    print res
+
-- 
GitLab