From 87c1c12ea752606eda2473e8b33fbfc99ea7153d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 21 Jul 2013 01:13:14 -0400
Subject: [PATCH] Shorten convolution tests
---
test/test_loopy.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/test/test_loopy.py b/test/test_loopy.py
index 39e883ce0..272962291 100644
--- a/test/test_loopy.py
+++ b/test/test_loopy.py
@@ -1187,6 +1187,7 @@ def test_convolution(ctx_factory):
def variant_1(knl):
knl = lp.split_iname(knl, "im_x", 16, inner_tag="l.0")
+ knl = lp.set_loop_priority(knl, "iimg,im_x_outer,im_y,ifeat,f_x,f_y")
return knl
def variant_2(knl):
@@ -1206,7 +1207,7 @@ def test_convolution(ctx_factory):
lp.auto_test_vs_ref(ref_knl, ctx, variant(knl),
parameters=dict(
im_w=128, im_h=128, f_w=f_w,
- nfeats=12, nimgs=17
+ nfeats=3, nimgs=3
))
@@ -1253,6 +1254,7 @@ def test_convolution_with_nonzero_base(ctx_factory):
def variant_1(knl):
knl = lp.split_iname(knl, "im_x", 16, inner_tag="l.0")
+ knl = lp.set_loop_priority(knl, "iimg,im_x_outer,im_y,ifeat,f_x,f_y")
return knl
for variant in [
--
GitLab