diff options
Diffstat (limited to 'src/tuning/copy_fast.cc')
| -rw-r--r-- | src/tuning/copy_fast.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/tuning/copy_fast.cc b/src/tuning/copy_fast.cc index 2da707be..09fdbaba 100644 --- a/src/tuning/copy_fast.cc +++ b/src/tuning/copy_fast.cc @@ -37,7 +37,7 @@ class TuneCopy { } // The list of arguments relevant for this routine - static std::vector<std::string> GetOptions() { return {kArgM, kArgN}; } + static std::vector<std::string> GetOptions() { return {kArgM, kArgN, kArgAlpha}; } // Tests for valid arguments static void TestValidArguments(const Arguments<T> &) { } @@ -86,9 +86,11 @@ class TuneCopy { std::vector<T> &, std::vector<T> &, std::vector<T> &a_mat, std::vector<T> &b_mat, std::vector<T> &, std::vector<T> &) { + auto alpha_buffer = std::vector<T>{args.alpha}; tuner.AddArgumentScalar(static_cast<int>(args.m)); tuner.AddArgumentInput(a_mat); tuner.AddArgumentOutput(b_mat); + tuner.AddArgumentInput(alpha_buffer); } // Describes how to compute the performance metrics |