Skip to content

Commit 8f738d0

Browse files
[TESTING] Enable execution of test_packed_8x8x32_resnet50 (#13799)
* [TESTING] Enable execution of test_packed_8x8x32_resnet50 Resnet 50 is passing now. Need to add below mentioned changes as well for test to pass. - Add condition for IndexMap - Add empty dict while calling tune_conv2d_template to avoid failure * Fix unused variable error * Remove unused variable and change check in if condition
1 parent 8f80e42 commit 8f738d0

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/meta_schedule/database/database_utils.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,8 @@ void JSONDumps(ObjectRef json_obj, std::ostringstream& os) {
7777
JSONDumps(kv.second, os);
7878
}
7979
os << "}";
80+
} else if (json_obj->IsInstance<tir::IndexMapNode>()) {
81+
// Do nothing for index maps to start
8082
} else {
8183
LOG(FATAL) << "TypeError: Unsupported type in JSON object: " << json_obj->GetTypeKey();
8284
}

tests/python/contrib/test_hexagon/metaschedule_e2e/test_resnet50_int8.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,7 @@ def test_packed_8x8x32_resnet50(hexagon_launcher):
410410

411411
if do_tune:
412412
hexagon_lowered = tune_conv2d_template(
413-
mod, _schedule_packed_8x8x32_conv2d, "packed_8x8x32", params, hexagon_launcher
413+
mod, _schedule_packed_8x8x32_conv2d, "packed_8x8x32", params, hexagon_launcher, {}
414414
)
415415
else:
416416
with tvm.transform.PassContext(opt_level=3):

0 commit comments

Comments
 (0)