Skip to content

Commit

Permalink
[MetaSchedule][ARM] Beautification of the function names (apache#14584)
Browse files Browse the repository at this point in the history
* [MetaSchedule][ARM] Enable ARM CPU intrinsic for MetaSchedule

* Beautification of the functions names
  • Loading branch information
dsbarinov1 authored Apr 12, 2023
1 parent 17f7db1 commit aee57f6
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/meta_schedule/schedule_rule/schedule_rule.cc
Original file line number Diff line number Diff line change
Expand Up @@ -295,7 +295,7 @@ Array<ScheduleRule> ScheduleRule::DefaultMicro() {
};
}

Array<ScheduleRule> GetNeonSpecificRules() {
Array<ScheduleRule> GetARMNeonSpecificRules() {
return {
ScheduleRule::MultiLevelTilingWithIntrin(
/*intrin_name=*/String("dot_4x4_i8i8s32_neon"),
Expand All @@ -311,7 +311,7 @@ Array<ScheduleRule> GetNeonSpecificRules() {
};
}

Array<ScheduleRule> GetDotprodSpecificRules() {
Array<ScheduleRule> GetARMDotprodSpecificRules() {
return {
ScheduleRule::MultiLevelTilingWithIntrin(
/*intrin_name=*/String("dot_4x4_i8i8s32_sdot"),
Expand Down Expand Up @@ -363,8 +363,8 @@ Array<ScheduleRule> ScheduleRule::DefaultARM(const String& type) {
ScheduleRule::AddRFactor(
/*max_jobs_per_core=*/8,
/*max_innermost_factor=*/Integer(32)),
"neon" == type ? GetNeonSpecificRules() : Array<ScheduleRule>{},
"dotprod" == type ? GetDotprodSpecificRules() : Array<ScheduleRule>{},
"neon" == type ? GetARMNeonSpecificRules() : Array<ScheduleRule>{},
"dotprod" == type ? GetARMDotprodSpecificRules() : Array<ScheduleRule>{},
ScheduleRule::MultiLevelTiling(
/*structure=*/"SSRSRS",
/*tile_binds=*/NullOpt,
Expand Down

0 comments on commit aee57f6

Please sign in to comment.