summaryrefslogtreecommitdiff
path: root/src/tuning/kernels/invert.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/tuning/kernels/invert.hpp')
-rw-r--r--src/tuning/kernels/invert.hpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/src/tuning/kernels/invert.hpp b/src/tuning/kernels/invert.hpp
index 0a0c9ce2..4f74674d 100644
--- a/src/tuning/kernels/invert.hpp
+++ b/src/tuning/kernels/invert.hpp
@@ -87,6 +87,15 @@ void InvertTestValidArguments(const int, const Arguments<T> &args) {
}
}
std::vector<Constraint> InvertSetConstraints(const int) { return {}; }
+template <typename T>
+LocalMemSizeInfo InvertComputeLocalMemSize(const int) {
+ return {
+ [] (std::vector<size_t> v) -> size_t {
+ return GetBytes(PrecisionValue<T>()) * (16 + v[0]) * 16;
+ },
+ {"LOCALPAD"}
+ };
+}
// Sets the kernel's arguments
template <typename T>