diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-12-31 18:49:12 +0100 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-12-31 18:49:12 +0100 |
commit | d9295250396eafe7d16c8730104f669cb66777e3 (patch) | |
tree | 5d4dc8eb0e1b78fcee0c62938607bee740c74a60 /src/tuning | |
parent | 153ac06cf262d2680d0152933156b1d1e15b3f86 (diff) |
Added support for the convgemm tuner in the tuner database
Diffstat (limited to 'src/tuning')
-rw-r--r-- | src/tuning/tuning.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/tuning/tuning.cpp b/src/tuning/tuning.cpp index f76af774..c5eee527 100644 --- a/src/tuning/tuning.cpp +++ b/src/tuning/tuning.cpp @@ -389,6 +389,12 @@ void Tuner(int argc, char* argv[], const int V, if (o == kArgAlpha) { metadata.push_back({"arg_alpha", ToString(args.alpha)}); } if (o == kArgBeta) { metadata.push_back({"arg_beta", ToString(args.beta)}); } if (o == kArgBatchCount) { metadata.push_back({"arg_batch_count", ToString(args.batch_count)}); } + if (o == kArgHeight) { metadata.push_back({"arg_height", ToString(args.height)}); } + if (o == kArgWidth) { metadata.push_back({"arg_width", ToString(args.width)}); } + if (o == kArgKernelH) { metadata.push_back({"arg_kernel_h", ToString(args.kernel_h)}); } + if (o == kArgKernelW) { metadata.push_back({"arg_kernel_w", ToString(args.kernel_w)}); } + if (o == kArgChannels) { metadata.push_back({"arg_channels", ToString(args.channels)}); } + if (o == kArgNumKernels) { metadata.push_back({"arg_num_kernels", ToString(args.num_kernels)}); } } PrintTimingsToFileAsJSON("clblast_" + settings.kernel_family + "_" + precision_string + ".json", device, platform, metadata, results); |