From 6d49fcadc7115d4c8d402dd622727c45ce90fab1 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 14 Mar 2016 11:21:21 -0500
Subject: [PATCH] Add Python sparse matvec example
---
examples/python/sparse.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
create mode 100644 examples/python/sparse.py
diff --git a/examples/python/sparse.py b/examples/python/sparse.py
new file mode 100644
index 000000000..14d157b6a
--- /dev/null
+++ b/examples/python/sparse.py
@@ -0,0 +1,18 @@
+import loopy as lp
+
+k = lp.make_kernel([
+ "[m] -> { [i] : 0 <= i < m }",
+ "[length] -> { [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
+ """,
+ [
+ lp.GlobalArg("colindices,values,x", shape=None),
+ "..."
+ ])
+print(k)
--
GitLab