From ba85a383fe523e462cb35245d3edb0bec39d60b7 Mon Sep 17 00:00:00 2001
From: Masahiro Hiramori <mhg00g13@gmail.com>
Date: Fri, 21 Jul 2023 19:11:56 +0900
Subject: [PATCH] [Target] Add Jetson Orin Nano tag

---
 src/target/tag.cc | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/target/tag.cc b/src/target/tag.cc
index 037d2e5937ca..28f762609b7d 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")},                                   \