diff options
Diffstat (limited to 'include/internal')
-rw-r--r-- | include/internal/routine.h | 2 | ||||
-rw-r--r-- | include/internal/utilities.h | 4 |
2 files changed, 6 insertions, 0 deletions
diff --git a/include/internal/routine.h b/include/internal/routine.h index 367917fd..c5b253b4 100644 --- a/include/internal/routine.h +++ b/include/internal/routine.h @@ -78,6 +78,8 @@ class Routine { const size_t inc, const size_t data_size); StatusCode TestVectorY(const size_t n, const Buffer<T> &buffer, const size_t offset, const size_t inc, const size_t data_size); + StatusCode TestVectorDot(const size_t n, const Buffer<T> &buffer, const size_t offset, + const size_t data_size); // Copies/transposes a matrix and padds/unpads it with zeroes. This method is also able to write // to symmetric and triangular matrices through optional arguments. diff --git a/include/internal/utilities.h b/include/internal/utilities.h index d9fdb9ab..466ac810 100644 --- a/include/internal/utilities.h +++ b/include/internal/utilities.h @@ -57,6 +57,7 @@ constexpr auto kArgCLeadDim = "ldc"; constexpr auto kArgAOffset = "offa"; constexpr auto kArgBOffset = "offb"; constexpr auto kArgCOffset = "offc"; +constexpr auto kArgDotOffset = "offdot"; constexpr auto kArgAlpha = "alpha"; constexpr auto kArgBeta = "beta"; @@ -105,6 +106,7 @@ struct Arguments { size_t a_offset = 0; size_t b_offset = 0; size_t c_offset = 0; + size_t dot_offset = 0; T alpha = T{1.0}; T beta = T{1.0}; size_t x_size = 1; @@ -112,6 +114,7 @@ struct Arguments { size_t a_size = 1; size_t b_size = 1; size_t c_size = 1; + size_t dot_size = 1; // Tuner-specific arguments double fraction = 1.0; // Client-specific arguments @@ -138,6 +141,7 @@ struct Buffers { Buffer<T> a_mat; Buffer<T> b_mat; Buffer<T> c_mat; + Buffer<T> dot; }; // ================================================================================================= |