From 5fa87e7c501ddaad5bedc4cfccad469f79e4456c Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 10 Sep 2011 15:32:05 -0500
Subject: [PATCH] Turn off slabs in plain matmul test.
---
test/test_matmul.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/test/test_matmul.py b/test/test_matmul.py
index c512640c4..194e71361 100644
--- a/test/test_matmul.py
+++ b/test/test_matmul.py
@@ -159,9 +159,11 @@ def test_plain_matrix_mul(ctx_factory):
],
name="matmul")
- knl = lp.split_dimension(knl, "i", 16, outer_tag="g.0", inner_tag="l.1")
- knl = lp.split_dimension(knl, "j", 16, outer_tag="g.1", inner_tag="l.0")
- knl = lp.split_dimension(knl, "k", 16)
+ knl = lp.split_dimension(knl, "i", 16,
+ outer_tag="g.0", inner_tag="l.1", no_slabs=True)
+ knl = lp.split_dimension(knl, "j", 16,
+ outer_tag="g.1", inner_tag="l.0", no_slabs=True)
+ knl = lp.split_dimension(knl, "k", 16, no_slabs=True)
knl = lp.add_prefetch(knl, 'a', ["k_inner", "i_inner"])
knl = lp.add_prefetch(knl, 'b', ["j_inner", "k_inner", ])
assert knl.get_problems({})[0] <= 2
--
GitLab