diff options
author | CNugteren <web@cedricnugteren.nl> | 2015-09-14 15:53:34 +0200 |
---|---|---|
committer | CNugteren <web@cedricnugteren.nl> | 2015-09-14 15:53:34 +0200 |
commit | 2a383f34501b386b8e6c4beb56c6ac694622f060 (patch) | |
tree | 782fa45888677b9fe0a1e12f85c6f795308935f2 /src/tuning/xgemv.cc | |
parent | e0c5312abb6f5b1c0e413fe69a8c7c46215cd3ae (diff) |
Added extra temporary buffer to tuners in preparation of Xdot routines
Diffstat (limited to 'src/tuning/xgemv.cc')
-rw-r--r-- | src/tuning/xgemv.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/tuning/xgemv.cc b/src/tuning/xgemv.cc index e22b5103..3d6fe595 100644 --- a/src/tuning/xgemv.cc +++ b/src/tuning/xgemv.cc @@ -56,6 +56,7 @@ class TuneXgemv { static size_t GetSizeA(const Arguments<T> &args) { return args.m * args.n; } static size_t GetSizeB(const Arguments<T> &) { return 1; } // N/A for this kernel static size_t GetSizeC(const Arguments<T> &) { return 1; } // N/A for this kernel + static size_t GetSizeTemp(const Arguments<T> &) { return 1; } // N/A for this kernel // Sets the tuning parameters and their possible values static void SetParameters(cltune::Tuner &tuner, const size_t id) { @@ -75,6 +76,7 @@ class TuneXgemv { // Sets the base thread configuration static std::vector<size_t> GlobalSize(const Arguments<T> &args) { return {args.m}; } + static std::vector<size_t> GlobalSizeRef(const Arguments<T> &args) { return GlobalSize(args); } static std::vector<size_t> LocalSize() { return {1}; } static std::vector<size_t> LocalSizeRef() { return {64}; } @@ -88,7 +90,8 @@ class TuneXgemv { // Sets the kernel's arguments static void SetArguments(cltune::Tuner &tuner, const Arguments<T> &args, std::vector<T> &x_vec, std::vector<T> &y_vec, - std::vector<T> &a_mat, std::vector<T> &, std::vector<T> &) { + std::vector<T> &a_mat, std::vector<T> &, std::vector<T> &, + std::vector<T> &) { auto a_rotated = (V==3) ? 1 : 0; tuner.AddArgumentScalar(static_cast<int>(args.m)); tuner.AddArgumentScalar(static_cast<int>(args.n)); |