From bfa28628289293858813e99328f57c384367306c Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 15 Dec 2012 18:08:47 -0500
Subject: [PATCH] Elementwise: Allow omitting zero for range slice base.
---
pyopencl/elementwise.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/pyopencl/elementwise.py b/pyopencl/elementwise.py
index 9f836095..60c4bc11 100644
--- a/pyopencl/elementwise.py
+++ b/pyopencl/elementwise.py
@@ -239,7 +239,10 @@ class ElementwiseKernel:
queue.device)
if range_ is not None:
- invocation_args.append(range_.start)
+ start = range_.start
+ if start is None:
+ start = 0
+ invocation_args.append(start)
invocation_args.append(range_.stop)
if range_.step is None:
step = 1
@@ -250,7 +253,7 @@ class ElementwiseKernel:
from pyopencl.array import splay
gs, ls = splay(queue,
- abs(range_.stop - range_.start)//step,
+ abs(range_.stop - start)//step,
max_wg_size)
else:
invocation_args.append(repr_vec.size)
--
GitLab