Skip to content

Commit fde75c1

Browse files
committed
Fix dtype
1 parent 4572190 commit fde75c1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/te/operation/create_primfunc.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ BlockRealize GenerateBlockFromTensors(const te::ComputeOp& compute_op,
165165
init_stmts.push_back(BufferStore(buffer, reduce->combiner->identity_element[i], indices));
166166
PrimExpr value{nullptr};
167167
if (n_buffers > 1) {
168-
temp_vars.push_back(Var("v_" + buffer->name, lhs[i].dtype()));
168+
temp_vars.push_back(Var("v_" + buffer->name, PrimType(lhs[i].dtype())));
169169
value = temp_vars.back();
170170
} else {
171171
value = reduce->combiner.get()->operator()(lhs, rhs)[i];

0 commit comments

Comments
 (0)