From e16afc9fcd703ec9d43290252da2baa3d316cb5a Mon Sep 17 00:00:00 2001 From: Ryoma Hiraoka <153697958+rym-h@users.noreply.github.com> Date: Thu, 14 Dec 2023 01:11:59 +0900 Subject: [PATCH] [Target] Add Jetson AGX Orin tags --- src/target/tag.cc | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/src/target/tag.cc b/src/target/tag.cc index e6521d384397..d8f341351d21 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{{"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{{"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")}, \