diff options
author | CNugteren <web@cedricnugteren.nl> | 2015-06-13 14:10:07 +0200 |
---|---|---|
committer | CNugteren <web@cedricnugteren.nl> | 2015-06-13 14:10:07 +0200 |
commit | 9b66883e9c016ed749e4e492416ac42b63a4ddd2 (patch) | |
tree | abd1fd196684160ec51f672e39051c263ab30b57 /src/tuning/xgemv.cc | |
parent | e522d1a74e6a877f32730da4807f54cf9a996679 (diff) |
Improved GEMV kernel with local memory and a tunable WPT
Diffstat (limited to 'src/tuning/xgemv.cc')
-rw-r--r-- | src/tuning/xgemv.cc | 6 |
1 files changed, 3 insertions, 3 deletions
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<T> &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<T> &args, tuner.AddArgumentScalar(0); tuner.AddArgumentInput(a_mat); tuner.AddArgumentScalar(0); - tuner.AddArgumentScalar(static_cast<int>(args.n)); + tuner.AddArgumentScalar(static_cast<int>(args.m)); tuner.AddArgumentInput(x_vec); tuner.AddArgumentScalar(0); tuner.AddArgumentScalar(1); |