From aee57f682ff3c6e9bb724426cb6b0a37076d9b14 Mon Sep 17 00:00:00 2001 From: dsbarinov1 <71228944+dsbarinov1@users.noreply.github.com> Date: Wed, 12 Apr 2023 13:34:14 +0300 Subject: [PATCH] [MetaSchedule][ARM] Beautification of the function names (#14584) * [MetaSchedule][ARM] Enable ARM CPU intrinsic for MetaSchedule * Beautification of the functions names --- src/meta_schedule/schedule_rule/schedule_rule.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/meta_schedule/schedule_rule/schedule_rule.cc b/src/meta_schedule/schedule_rule/schedule_rule.cc index 35f1151c9c1d..99640a6e7ffd 100644 --- a/src/meta_schedule/schedule_rule/schedule_rule.cc +++ b/src/meta_schedule/schedule_rule/schedule_rule.cc @@ -295,7 +295,7 @@ Array ScheduleRule::DefaultMicro() { }; } -Array GetNeonSpecificRules() { +Array GetARMNeonSpecificRules() { return { ScheduleRule::MultiLevelTilingWithIntrin( /*intrin_name=*/String("dot_4x4_i8i8s32_neon"), @@ -311,7 +311,7 @@ Array GetNeonSpecificRules() { }; } -Array GetDotprodSpecificRules() { +Array GetARMDotprodSpecificRules() { return { ScheduleRule::MultiLevelTilingWithIntrin( /*intrin_name=*/String("dot_4x4_i8i8s32_sdot"), @@ -363,8 +363,8 @@ Array ScheduleRule::DefaultARM(const String& type) { ScheduleRule::AddRFactor( /*max_jobs_per_core=*/8, /*max_innermost_factor=*/Integer(32)), - "neon" == type ? GetNeonSpecificRules() : Array{}, - "dotprod" == type ? GetDotprodSpecificRules() : Array{}, + "neon" == type ? GetARMNeonSpecificRules() : Array{}, + "dotprod" == type ? GetARMDotprodSpecificRules() : Array{}, ScheduleRule::MultiLevelTiling( /*structure=*/"SSRSRS", /*tile_binds=*/NullOpt,