From a98e49126e3a1aa40b6959bf13d421fee5e0d5e8 Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Wed, 5 Apr 2023 19:43:45 +0900 Subject: [PATCH] [Target] Fix Jetson AGX Xavier CPU core count --- src/target/tag.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/target/tag.cc b/src/target/tag.cc index 34a939183be4..dd4d5b1d0abc 100644 --- a/src/target/tag.cc +++ b/src/target/tag.cc @@ -92,7 +92,7 @@ TVM_REGISTER_TARGET_TAG("nvidia/jetson-agx-xavier") {"host", Map{{"kind", String("llvm")}, {"mtriple", String("aarch64-linux-gnu")}, {"mcpu", String("carmel")}, - {"num-cores", Integer(4)}}}}); + {"num-cores", Integer(8)}}}}); #define TVM_REGISTER_CUDA_TAG(Name, Arch, SharedMem, RegPerBlock) \ TVM_REGISTER_TARGET_TAG(Name).set_config({ \