From a08d9aa404f2c2ec858632e30640ce99bf94310b Mon Sep 17 00:00:00 2001 From: Kathryn-cat Date: Sat, 6 Aug 2022 14:11:55 -0700 Subject: [PATCH 1/2] minor bug fix --- src/meta_schedule/feature_extractor/per_store_feature.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/meta_schedule/feature_extractor/per_store_feature.cc b/src/meta_schedule/feature_extractor/per_store_feature.cc index c29e5d61f0bb..3563eafa939b 100644 --- a/src/meta_schedule/feature_extractor/per_store_feature.cc +++ b/src/meta_schedule/feature_extractor/per_store_feature.cc @@ -885,7 +885,7 @@ void Feature::SubFeature::SetStride(const LoopNest& loop_nest, arith::Analyzer* // Calculate this->prod int64_t& prod = this->prod_non_strided_loop_extent = 1; for (int j = n_loops - 1; j > i; --j) { - if (const int64_t* extent = GetLoopIntExtent(loops[n_loops - 1])) { + if (const int64_t* extent = GetLoopIntExtent(loops[j])) { prod *= *extent; } } From cf3bd7794e979112efa425b9c3a62d17af2f600c Mon Sep 17 00:00:00 2001 From: Kathryn-cat Date: Mon, 8 Aug 2022 18:19:45 -0700 Subject: [PATCH 2/2] small tix --- src/meta_schedule/feature_extractor/per_store_feature.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/meta_schedule/feature_extractor/per_store_feature.cc b/src/meta_schedule/feature_extractor/per_store_feature.cc index 3563eafa939b..571bb2face00 100644 --- a/src/meta_schedule/feature_extractor/per_store_feature.cc +++ b/src/meta_schedule/feature_extractor/per_store_feature.cc @@ -894,7 +894,7 @@ void Feature::SubFeature::SetStride(const LoopNest& loop_nest, arith::Analyzer* void Feature::SubFeature::SetReuse(const LoopNest& loop_nest, int64_t top_loop_touch_bytes, const ForBufferMap& buffer_touched_under_loop) { const BufferNode* buffer = this->buffer; - // Step 0. Collect all `Var`s that appears in the buffer region + // Step 3.1. Collect all `Var`s that appears in the buffer region std::unordered_set region_vars; for (const MultiIndex& multi_index : this->multi_indices) { for (const PrimExpr& index : multi_index) {