From ae300774267446fed612d3e6a117a51f71b7ae79 Mon Sep 17 00:00:00 2001
From: Prakalp Srivastava <prakalp@octoml.ai>
Date: Mon, 3 Apr 2023 09:32:44 -0700
Subject: [PATCH] [Target] Add A10G gpu cuda tag

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

diff --git a/src/target/tag.cc b/src/target/tag.cc
index 34a939183be4..ded4855b0456 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);