[tvm] branch main updated: [Target] Add A10G gpu cuda tag (#14467)

[tqchen]

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

tqchen 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 4d59c959aa [Target] Add A10G gpu cuda tag (#14467)
4d59c959aa is described below

commit 4d59c959aa5a345fbad78da5e9211c89f265c2ad
Author: Prakalp Srivastava <prak...@octoml.ai>
AuthorDate: Fri Apr 7 21:34:06 2023 -0400

    [Target] Add A10G gpu cuda tag (#14467)
---
 src/target/tag.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/target/tag.cc b/src/target/tag.cc
index 34a939183b..ded4855b04 100644
--- a/src/target/tag.cc
+++ b/src/target/tag.cc
@@ -115,6 +115,7 @@ TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a100", "sm_80", 49152, 
65536);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a40", "sm_86", 49152, 65536);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a30", "sm_80", 49152, 65536);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a10", "sm_86", 49152, 65536);
+TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a10g", "sm_86", 49152, 65536);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a16", "sm_86", 49152, 65536);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a2", "sm_86", 49152, 65536);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-t4", "sm_75", 49152, 65536);