Hzfengsy commented on code in PR #18079:
URL: https://github.com/apache/tvm/pull/18079#discussion_r2157971010
##########
src/script/ir_builder/tir/ir.cc:
##########
@@ -219,12 +219,47 @@ void Writes(Array<ObjectRef> buffer_slices) {
frame->writes = writes;
}
+/*! \brief Recursively merge two annotations, the new attrs will override the
old ones */
+Map<String, Any> MergeAnnotations(const Map<String, Any>& new_attrs,
+ const Map<String, Any>& old_attrs) {
+ Map<String, Any> result = old_attrs;
+ for (const auto& [key, value] : new_attrs) {
+ auto old_value = old_attrs.Get(key);
+ // Case 1: the key is not in the old annotations, set the key to the new
value
+ if (!old_value) {
+ result.Set(key, value);
+ continue;
+ }
+
+ // Case 2: the key is in the old annotations
+ // Case 2.1: both are dicts
+ auto old_dict = old_value->try_cast<Map<String, Any>>();
+ auto new_dict = value.try_cast<Map<String, Any>>();
+ if (old_dict && new_dict) {
+ // Recursively merge the two dicts
+ auto merged_dict = MergeAnnotations(*old_dict, *new_dict);
+ result.Set(key, merged_dict);
+ continue;
+ }
+ // Case 2.2: the values are not both dicts, check if the keys are the same
+ if (!ffi::AnyEqual()(old_value.value(), value)) {
+ LOG(FATAL) << "ValueError: Try to merge two annotations with different
values for key `"
Review Comment:
Also I added a test case to test the error in
tests/python/tvmscript/test_tvmscript_error_report.py
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
