Fix dtype

apache · Mar 18, 2022 · fde75c1 · fde75c1
1 parent 4572190
commit fde75c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/te/operation/create_primfunc.cc b/src/te/operation/create_primfunc.cc
@@ -165,7 +165,7 @@ BlockRealize GenerateBlockFromTensors(const te::ComputeOp& compute_op,
       init_stmts.push_back(BufferStore(buffer, reduce->combiner->identity_element[i], indices));
       PrimExpr value{nullptr};
       if (n_buffers > 1) {
-        temp_vars.push_back(Var("v_" + buffer->name, lhs[i].dtype()));
+        temp_vars.push_back(Var("v_" + buffer->name, PrimType(lhs[i].dtype())));
         value = temp_vars.back();
       } else {
         value = reduce->combiner.get()->operator()(lhs, rhs)[i];