[tvm] branch main updated: [Target] Add Jetson Orin Nano tag (#15380)

[ruihangl]

This is an automated email from the ASF dual-hosted git repository.

ruihangl 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 e4af30220f [Target] Add Jetson Orin Nano tag (#15380)
e4af30220f is described below

commit e4af30220f34a42cc3f0a9f4a564bb72c31de3c4
Author: Masahiro Hiramori <mhg00...@gmail.com>
AuthorDate: Fri Jul 21 17:17:28 2023 -0700

    [Target] Add Jetson Orin Nano tag (#15380)
    
    This PR adds the `nvidia/jetson-orin-nano` target tag.
---
 src/target/tag.cc | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/target/tag.cc b/src/target/tag.cc
index 037d2e5937..28f762609b 100644
--- a/src/target/tag.cc
+++ b/src/target/tag.cc
@@ -94,6 +94,18 @@ TVM_REGISTER_TARGET_TAG("nvidia/jetson-agx-xavier")
                                                  {"mcpu", String("carmel")},
                                                  {"num-cores", Integer(8)}}}});
 
+TVM_REGISTER_TARGET_TAG("nvidia/jetson-orin-nano")
+    .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("carmel")},
+                                                 {"num-cores", Integer(6)}}}});
+
 #define TVM_REGISTER_CUDA_TAG(Name, Arch, SharedMem, RegPerBlock) \
   TVM_REGISTER_TARGET_TAG(Name).set_config({                      \
       {"kind", String("cuda")},                                   \