This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new aee57f682f [MetaSchedule][ARM] Beautification of the function names
(#14584)
aee57f682f is described below
commit aee57f682ff3c6e9bb724426cb6b0a37076d9b14
Author: dsbarinov1 <[email protected]>
AuthorDate: Wed Apr 12 13:34:14 2023 +0300
[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 35f1151c9c..99640a6e7f 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> ScheduleRule::DefaultMicro() {
};
}
-Array<ScheduleRule> GetNeonSpecificRules() {
+Array<ScheduleRule> GetARMNeonSpecificRules() {
return {
ScheduleRule::MultiLevelTilingWithIntrin(
/*intrin_name=*/String("dot_4x4_i8i8s32_neon"),
@@ -311,7 +311,7 @@ Array<ScheduleRule> GetNeonSpecificRules() {
};
}
-Array<ScheduleRule> GetDotprodSpecificRules() {
+Array<ScheduleRule> GetARMDotprodSpecificRules() {
return {
ScheduleRule::MultiLevelTilingWithIntrin(
/*intrin_name=*/String("dot_4x4_i8i8s32_sdot"),
@@ -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,