diff --git a/src/meta_schedule/space_generator/space_generator.cc b/src/meta_schedule/space_generator/space_generator.cc index bd124511b83c..cb89b3b817af 100644 --- a/src/meta_schedule/space_generator/space_generator.cc +++ b/src/meta_schedule/space_generator/space_generator.cc @@ -25,7 +25,7 @@ String GetRuleKindFromTarget(const Target& target) { if (target->kind->name == "llvm") { static const PackedFunc* f_check_vnni = runtime::Registry::Get("tvm.topi.x86.utils.target_has_vnni"); - ICHECK(*f_check_vnni != nullptr) << "The `target_has_vnni` func is not in tvm registry."; + ICHECK(f_check_vnni != nullptr) << "The `target_has_vnni` func is not in tvm registry."; if (target->GetAttr("mcpu") && (*f_check_vnni)(target->GetAttr("mcpu").value())) { return "vnni"; diff --git a/tests/python/unittest/test_meta_schedule_space_cpu_winograd.py b/tests/python/unittest/test_meta_schedule_space_cpu_winograd.py index 78b75d592ed4..135304286b4b 100644 --- a/tests/python/unittest/test_meta_schedule_space_cpu_winograd.py +++ b/tests/python/unittest/test_meta_schedule_space_cpu_winograd.py @@ -19,7 +19,6 @@ from tvm.meta_schedule.testing.space_generation import ( check_sketches, generate_design_space, - print_sketches, ) from tvm.meta_schedule.testing.te_workload import create_te_workload from tvm.script import tir as T