From b21bd09162488745a5079a6c7d4d6f3e6e9347fb Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 11 Apr 2016 17:46:18 -0500
Subject: [PATCH] Make Python sparse example work

---
 examples/python/sparse.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/examples/python/sparse.py b/examples/python/sparse.py
index 03f1ad22e..bb143d62b 100644
--- a/examples/python/sparse.py
+++ b/examples/python/sparse.py
@@ -1,14 +1,17 @@
 import loopy as lp
+import numpy as np
 
 k = lp.make_kernel([
     "{ [i] : 0 <= i < m }",
     "{ [j] : 0 <= j < length }"],
     """
-    rowstart = rowstarts[i]
-    rowend = rowstarts[1 + i]
-    length = rowend + (-1)*rowstart
-    rowsum = 0 {id=zerosum}
-    rowsum = rowsum + x[-1 + colindices[-1 + rowstart + j]]*values[-1 + rowstart + j] {dep=zerosum}
-    y[i] = rowsum
+    <> rowstart = rowstarts[i]
+    <> rowend = rowstarts[i]
+    <> length = rowend - rowstart
+    y[i] = sum(j, values[rowstart+j] * x[colindices[rowstart + j]])
     """)
-print(k)
+
+k = lp.add_and_infer_dtypes(k, {
+    "values,x": np.float64, "rowstarts,colindices": k.index_dtype
+    })
+print(lp.generate_code(k)[0])
-- 
GitLab