This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch refactor-s2 in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 250cc30d114f554da9628a63a72e96c9d2e7e728 Author: tqchen <[email protected]> AuthorDate: Sat May 3 09:06:17 2025 -0400 Update to reflect latest variant api --- src/support/ffi_testing.cc | 2 +- src/tir/schedule/primitive/block_annotate.cc | 2 +- src/tir/transforms/compact_buffer_region.cc | 6 +++--- src/tir/transforms/ir_utils.cc | 4 ++-- 4 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/support/ffi_testing.cc b/src/support/ffi_testing.cc index 87447d1b5f..0b855e5e83 100644 --- a/src/support/ffi_testing.cc +++ b/src/support/ffi_testing.cc @@ -179,7 +179,7 @@ TVM_REGISTER_GLOBAL("testing.AcceptsVariant") if (auto opt_str = arg.as<String>()) { return opt_str.value()->GetTypeKey(); } else { - return arg.Get<Integer>()->GetTypeKey(); + return arg.get<Integer>()->GetTypeKey(); } }); diff --git a/src/tir/schedule/primitive/block_annotate.cc b/src/tir/schedule/primitive/block_annotate.cc index 22fc716f5e..97154d189c 100644 --- a/src/tir/schedule/primitive/block_annotate.cc +++ b/src/tir/schedule/primitive/block_annotate.cc @@ -238,7 +238,7 @@ void StorageAlign(ScheduleState self, const StmtSRef& block_sref, int buffer_ind StorageAlignTuple new_storage_align_tuple{buffer_index, axis, factor, offset}; for (size_t j = 0; j < storage_align_annotation.size(); ++j) { const auto& storage_align_tuple = storage_align_annotation[j]; - if (storage_align_tuple.Get<0>() == buffer_index && storage_align_tuple.Get<1>() == axis) { + if (storage_align_tuple.get<0>() == buffer_index && storage_align_tuple.get<1>() == axis) { storage_align_annotation.Set(j, std::move(new_storage_align_tuple)); found = true; break; diff --git a/src/tir/transforms/compact_buffer_region.cc b/src/tir/transforms/compact_buffer_region.cc index 0973dfade7..b18c076510 100644 --- a/src/tir/transforms/compact_buffer_region.cc +++ b/src/tir/transforms/compact_buffer_region.cc @@ -716,9 +716,9 @@ Stmt BufferCompactorCompact( if (it != storage_align.end()) { std::vector<DimAlignInfo> dim_aligns(buffer->shape.size()); for (const StorageAlignTuple& dim_align : (*it).second) { - int dim = dim_align.Get<1>(); - int factor = dim_align.Get<2>(); - int offset = dim_align.Get<3>(); + int dim = dim_align.get<1>(); + int factor = dim_align.get<2>(); + int offset = dim_align.get<3>(); dim_aligns.at(dim) = {factor, offset}; } alloc_info.dim_aligns = std::move(dim_aligns); diff --git a/src/tir/transforms/ir_utils.cc b/src/tir/transforms/ir_utils.cc index bd367e5056..3664abcf56 100644 --- a/src/tir/transforms/ir_utils.cc +++ b/src/tir/transforms/ir_utils.cc @@ -752,7 +752,7 @@ class StorageAlignCollector : public StmtVisitor { if (it != op->annotations.end()) { auto storage_align_annotation = Downcast<StorageAlignAnnotation>((*it).second); for (const auto& storage_align_tuple : storage_align_annotation) { - int buffer_index = storage_align_tuple.Get<0>(); + int buffer_index = storage_align_tuple.get<0>(); const Buffer& buffer = op->writes[buffer_index]->buffer; storage_align_[buffer->data].push_back(storage_align_tuple); } @@ -766,7 +766,7 @@ class StorageAlignCollector : public StmtVisitor { if (it != op->annotations.end()) { auto storage_align_annotation = Downcast<StorageAlignAnnotation>((*it).second); for (const auto& storage_align_tuple : storage_align_annotation) { - int buffer_index = storage_align_tuple.Get<0>(); + int buffer_index = storage_align_tuple.get<0>(); // the first buffer idx info is meaningless for allocate // stmt and should set as negative intentionally. ICHECK_EQ(buffer_index, -1);
