(tvm) branch main updated: Add NVIDIA Hopper H100 target tag (#16407)

[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 3053f65da7 Add NVIDIA Hopper H100 target tag (#16407)
3053f65da7 is described below

commit 3053f65da7b156763fdb3110835947113b327b39
Author: Andrey Malyshev <elvin.n...@gmail.com>
AuthorDate: Tue Jan 16 17:00:30 2024 +0200

    Add NVIDIA Hopper H100 target tag (#16407)
    
    Co-authored-by: Andrey Malyshev <sunnyppa...@gmail.com>
---
 src/target/tag.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/target/tag.cc b/src/target/tag.cc
index d8f341351d..9caeec3b92 100644
--- a/src/target/tag.cc
+++ b/src/target/tag.cc
@@ -155,6 +155,8 @@ TVM_REGISTER_CUDA_TAG("nvidia/tesla-c2050", "sm_20", 49152, 
32768);
 TVM_REGISTER_CUDA_TAG("nvidia/tesla-c2070", "sm_20", 49152, 32768);
 TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a100", "sm_80", 49152, 65536)
     .with_config("l2_cache_size_bytes", Integer(41943040));
+TVM_REGISTER_CUDA_TAG("nvidia/nvidia-h100", "sm_90", 49152, 65536)
+    .with_config("l2_cache_size_bytes", Integer(52428800));
 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);