[general] a bunch of fixes in anticipation of proper triton vs cudnn

benchmarks

* DNN: Added partial auto-tuning mode and skeleton for heuristics
* Examples: Moduralized benchmarking and now evaluating ResNet-18 shapes
This commit is contained in:
Philippe Tillet
2019-07-21 20:17:56 -07:00
parent b1d81a5802
commit ead368d1ed
10 changed files with 221 additions and 147 deletions

View File

@@ -1,5 +1,6 @@
#include <sstream>
#include "triton/dnn/shift.h"
#include "triton/dnn/heuristics.h"
#include "triton/tools/bench.hpp"
namespace triton{
@@ -513,12 +514,14 @@ else{
}
// small search space for partial auto-tuning
std::vector<params_t> shift::search_space() const {
return dot_search_space(AT_, BT_);
}
// simple parameter heuristics
std::vector<unsigned> shift::default_params() const {
typedef std::vector<unsigned> params_t;
std::map<std::tuple<op_t, size_t, size_t>, params_t> params = {
{{}, {}}
};
params_t shift::heuristics() const {
return dot_heuristics(AT_, BT_, M_, N_, K_);
}