[dnn/conv] added triton-c code for wgrad

2019-05-11 18:09:23 -04:00
parent fc4daf11dd
commit f6fe9492e4
3 changed files with 299 additions and 173 deletions
--- a/lib/codegen/selection.cpp
+++ b/lib/codegen/selection.cpp
@@ -812,7 +812,9 @@ void selection::lower_tile_instruction(ir::instruction *ins, llvm::IRBuilder<> &
              std::swap(b_idx[0], b_idx[1]);
            Value *a = TA->get_value(a_idx);
            Value *b = TB->get_value(b_idx);
+//            res = builder.CreateCall(f_mul_add, {ConstantFP::get(a->getType(), 1), ConstantFP::get(b->getType(), 1), res});
            res = builder.CreateCall(f_mul_add, {a, b, res});
+
          }
          result->set_value(idx, res);
        });