Skip to content

Commit

Permalink
[Target] Fix Jetson AGX Xavier CPU core count (#14508)
Browse files Browse the repository at this point in the history
  • Loading branch information
mshr-h authored Apr 8, 2023
1 parent 2a23d59 commit f990c0b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/target/tag.cc
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ TVM_REGISTER_TARGET_TAG("nvidia/jetson-agx-xavier")
{"host", Map<String, ObjectRef>{{"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({ \
Expand Down

0 comments on commit f990c0b

Please sign in to comment.