From 9b66883e9c016ed749e4e492416ac42b63a4ddd2 Mon Sep 17 00:00:00 2001 From: CNugteren Date: Sat, 13 Jun 2015 14:10:07 +0200 Subject: Improved GEMV kernel with local memory and a tunable WPT --- src/tuning/xgemv.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/tuning') diff --git a/src/tuning/xgemv.cc b/src/tuning/xgemv.cc index 74bb77a5..6037a5a0 100644 --- a/src/tuning/xgemv.cc +++ b/src/tuning/xgemv.cc @@ -37,8 +37,8 @@ void XgemvTune(const Arguments &args, tuner.SetReferenceFromString(sources, "Xgemv", {args.m}, {64}); // Sets the tunable parameters and their possible values - tuner.AddParameter(id, "WGS", {64, 128}); - tuner.AddParameter(id, "WPT", {1}); + tuner.AddParameter(id, "WGS", {64, 128, 256, 512, 1024, 1536, 2048}); + tuner.AddParameter(id, "WPT", {1, 2, 4}); tuner.AddParameter(id, "VW", {1}); // Tests for a specific precision @@ -58,7 +58,7 @@ void XgemvTune(const Arguments &args, tuner.AddArgumentScalar(0); tuner.AddArgumentInput(a_mat); tuner.AddArgumentScalar(0); - tuner.AddArgumentScalar(static_cast(args.n)); + tuner.AddArgumentScalar(static_cast(args.m)); tuner.AddArgumentInput(x_vec); tuner.AddArgumentScalar(0); tuner.AddArgumentScalar(1); -- cgit v1.2.3