diff --git a/examples/demo_meta_codepy.py b/examples/demo_meta_codepy.py index 01c31082a985e82aae82aff9c24aaf350c62a322..c2838da00e182ec3d060b2f1f8c880f76f523725 100644 --- a/examples/demo_meta_codepy.py +++ b/examples/demo_meta_codepy.py @@ -34,8 +34,8 @@ mod = Module([ for name in ["tgt", "op1", "op2"]]))), Block([ Initializer(POD(numpy.int32, "idx"), - "get_local_id(0) + %d * %d * get_group_id(0)" - % (local_size, thread_strides)) + "get_local_id(0) + %d * get_group_id(0)" + % (local_size*thread_strides)) ]+[ Assign( "tgt[idx+%d]" % (o*local_size),