[dnn/blocksparse] added heuristics for block-sparse dot

2019-07-31 17:12:36 -07:00
parent bb32ac56c9
commit f7bd976fc7
4 changed files with 32 additions and 4 deletions
--- a/lib/codegen/tune.cpp
+++ b/lib/codegen/tune.cpp
@@ -228,7 +228,7 @@ void tune::run(ir::module &mod) {
        nts->set_value(1);
      }
      else {
-        ir::metaparameter *fpw = ir::metaparameter::create(ctx, ty, 2, 2);
+        ir::metaparameter *fpw = ir::metaparameter::create(ctx, ty, 1, 4);
        ir::metaparameter *wpt = ir::metaparameter::create(ctx, ty, 1, 4);
        connected_components(node, {fpw, wpt}, {"fpw", "wpt"}, nodes_, dependencies_, group_id++);
      }