summaryrefslogtreecommitdiff
path: root/src/tuning/tuning.hpp
diff options
context:
space:
mode:
authorCedric Nugteren <web@cedricnugteren.nl>2018-07-29 15:07:51 +0200
committerKoichi Akabe <vbkaisetsu@gmail.com>2018-12-18 13:59:26 +0900
commit1f0cd618247971f0803aff94ddf795d9e3d19428 (patch)
tree7a0a01742d59f633937c3b6dd760a9e51a69c9bf /src/tuning/tuning.hpp
parent301dc280dfe75ff3c8b219f64aea83a6bf2f0c8d (diff)
Added first version of a tuner for the ConvGemm direct kernel
Diffstat (limited to 'src/tuning/tuning.hpp')
-rw-r--r--src/tuning/tuning.hpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/tuning/tuning.hpp b/src/tuning/tuning.hpp
index 37a042ff..61cc3bda 100644
--- a/src/tuning/tuning.hpp
+++ b/src/tuning/tuning.hpp
@@ -41,6 +41,13 @@ struct TunerDefaults {
size_t default_m = 1;
size_t default_n = 1;
size_t default_k = 1;
+ size_t channels = 1;
+ size_t height = 1;
+ size_t width = 1;
+ size_t kernel_h = 3;
+ size_t kernel_w = 3;
+ size_t num_kernels = 1;
+ size_t batch_count = 1;
// Other defaults
size_t default_batch_count = 1;