diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-07-29 15:07:51 +0200 |
---|---|---|
committer | Koichi Akabe <vbkaisetsu@gmail.com> | 2018-12-18 13:59:26 +0900 |
commit | 1f0cd618247971f0803aff94ddf795d9e3d19428 (patch) | |
tree | 7a0a01742d59f633937c3b6dd760a9e51a69c9bf /src/tuning/tuning.hpp | |
parent | 301dc280dfe75ff3c8b219f64aea83a6bf2f0c8d (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.hpp | 7 |
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; |