Skip to content

Commit

Permalink
reformat
Browse files Browse the repository at this point in the history
  • Loading branch information
MasterJH5574 committed May 15, 2021
1 parent dee0b86 commit f521aff
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions src/tir/schedule/primitives/reduction.cc
Original file line number Diff line number Diff line change
Expand Up @@ -228,10 +228,10 @@ StmtSRef RFactor(ScheduleState self, const StmtSRef& loop_sref, int factor_axis)
rf_buf.CopyOnWrite()->shape = rf_shape;
rf_buf.CopyOnWrite()->name = rf_buf->name + ".rf";
rf_buf.CopyOnWrite()->data = rf_buf->data.copy_with_suffix(".rf");
BufferStore rf_update(
rf_buf,
reducer.value().get()->operator()({BufferLoad(rf_buf, rf_indices)}, {combiner_rhs.value()})[0],
rf_indices);
BufferStore rf_update(rf_buf,
reducer.value().get()->operator()({BufferLoad(rf_buf, rf_indices)},
{combiner_rhs.value()})[0],
rf_indices);
auto f_rf_create_rw_region = [&](const Array<BufferRegion>& regions) {
Array<BufferRegion> new_regions;
new_regions.reserve(static_cast<int64_t>(regions.size()));
Expand All @@ -253,7 +253,7 @@ StmtSRef RFactor(ScheduleState self, const StmtSRef& loop_sref, int factor_axis)
/*name_hint=*/block->name_hint + "_rf",
/*body=*/Substitute(static_cast<Stmt>(rf_update), var_map),
/*init=*/
Substitute(static_cast<Stmt>(BufferStore(rf_buf, init->value, rf_indices)), var_map));
Substitute(static_cast<Stmt>(BufferStore(rf_buf, init->value, rf_indices)), var_map));
BlockRealize rf_block_realize(rf_bindings, block_realize->predicate, rf_block);
// Finish constructing the rfactor block.

Expand Down

0 comments on commit f521aff

Please sign in to comment.