Fix the Type bug in ConvertSSA. (apache#6709)

Co-authored-by: YushengMa <yusheng.ma@streamcomputing.com>
jroesch · Oct 19, 2020 · 28e9ab7 · 28e9ab7
1 parent cc89b76
commit 28e9ab7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tir/transforms/ir_utils.cc b/src/tir/transforms/ir_utils.cc
@@ -158,7 +158,7 @@ class IRConvertSSA final : public StmtExprMutator {
   Stmt VisitStmt_(const AllocateNode* op) final {
     const Var& v = op->buffer_var;
     if (defined_.count(v.get())) {
-      Var new_var(v->name_hint, v.dtype());
+      Var new_var(v->name_hint, v->type_annotation);
       scope_[v.get()].push_back(new_var);
       Stmt stmt = StmtExprMutator::VisitStmt_(op);
       scope_[v.get()].pop_back();