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 37c38acc46 [Target] Add Jetson AGX Orin tags (#16231)
37c38acc46 is described below
commit 37c38acc46409647762d7b6181172ac83eef9bd2
Author: rym-h <[email protected]>
AuthorDate: Fri Dec 15 16:21:11 2023 +0900
[Target] Add Jetson AGX Orin tags (#16231)
---
src/target/tag.cc | 24 ++++++++++++++++++++++++
1 file changed, 24 insertions(+)
diff --git a/src/target/tag.cc b/src/target/tag.cc
index e6521d3843..d8f341351d 100644
--- a/src/target/tag.cc
+++ b/src/target/tag.cc
@@ -106,6 +106,30 @@ TVM_REGISTER_TARGET_TAG("nvidia/jetson-orin-nano")
{"mcpu", String("carmel")},
{"num-cores", Integer(6)}}}});
+TVM_REGISTER_TARGET_TAG("nvidia/jetson-agx-orin-32gb")
+ .set_config({{"kind", String("cuda")},
+ {"arch", String("sm_87")},
+ {"max_shared_memory_per_block", Integer(49152)},
+ {"max_threads_per_block", Integer(1024)},
+ {"thread_warp_size", Integer(32)},
+ {"registers_per_block", Integer(65536)},
+ {"host", Map<String, ObjectRef>{{"kind", String("llvm")},
+ {"mtriple",
String("aarch64-linux-gnu")},
+ {"mcpu",
String("cortex-a78")},
+ {"num-cores", Integer(8)}}}});
+
+TVM_REGISTER_TARGET_TAG("nvidia/jetson-agx-orin-64gb")
+ .set_config({{"kind", String("cuda")},
+ {"arch", String("sm_87")},
+ {"max_shared_memory_per_block", Integer(49152)},
+ {"max_threads_per_block", Integer(1024)},
+ {"thread_warp_size", Integer(32)},
+ {"registers_per_block", Integer(65536)},
+ {"host", Map<String, ObjectRef>{{"kind", String("llvm")},
+ {"mtriple",
String("aarch64-linux-gnu")},
+ {"mcpu",
String("cortex-a78")},
+ {"num-cores",
Integer(12)}}}});
+
#define TVM_REGISTER_CUDA_TAG(Name, Arch, SharedMem, RegPerBlock) \
TVM_REGISTER_TARGET_TAG(Name).set_config({ \
{"kind", String("cuda")}, \
