Skip to content

Commit 3053f65

Browse files
elvin-ngalka5
andauthored
Add NVIDIA Hopper H100 target tag (#16407)
Co-authored-by: Andrey Malyshev <[email protected]>
1 parent 68be158 commit 3053f65

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/target/tag.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,8 @@ TVM_REGISTER_CUDA_TAG("nvidia/tesla-c2050", "sm_20", 49152, 32768);
155155
TVM_REGISTER_CUDA_TAG("nvidia/tesla-c2070", "sm_20", 49152, 32768);
156156
TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a100", "sm_80", 49152, 65536)
157157
.with_config("l2_cache_size_bytes", Integer(41943040));
158+
TVM_REGISTER_CUDA_TAG("nvidia/nvidia-h100", "sm_90", 49152, 65536)
159+
.with_config("l2_cache_size_bytes", Integer(52428800));
158160
TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a40", "sm_86", 49152, 65536);
159161
TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a30", "sm_80", 49152, 65536);
160162
TVM_REGISTER_CUDA_TAG("nvidia/nvidia-a10", "sm_86", 49152, 65536);

0 commit comments

Comments
 (0)