From 708c49c4647e568a2099e446ba6e928d61b3d99d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 12 Aug 2015 14:41:45 -0500
Subject: [PATCH] Add another fusion test
---
test/test_loopy.py | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git a/test/test_loopy.py b/test/test_loopy.py
index 874bb55fb..87b4a95ad 100644
--- a/test/test_loopy.py
+++ b/test/test_loopy.py
@@ -2127,6 +2127,22 @@ def test_alias_temporaries(ctx_factory):
parameters=dict(n=30))
+def test_fusion():
+ exp_kernel = lp.make_kernel(
+ ''' { [i]: 0<=i<n } ''',
+ ''' exp[i] = pow(E, z[i])''',
+ assumptions="n>0")
+
+ sum_kernel = lp.make_kernel(
+ '{ [j]: 0<=j<n }',
+ 'out2 = sum(j, exp[j])',
+ assumptions='n>0')
+
+ knl = lp.fuse_kernels([exp_kernel, sum_kernel])
+
+ print(knl)
+
+
if __name__ == "__main__":
if len(sys.argv) > 1:
exec(sys.argv[1])
--
GitLab