From 825ecdab84f2c524a0676042f33c32daa5cd9ea1 Mon Sep 17 00:00:00 2001 From: thenumberouscode Date: Fri, 15 Aug 2025 17:18:22 +0800 Subject: [PATCH] add target nvidia rtx 5060ti --- src/target/tag.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/target/tag.cc b/src/target/tag.cc index 6bc3c07cbff1..f305c84e09a4 100644 --- a/src/target/tag.cc +++ b/src/target/tag.cc @@ -271,6 +271,8 @@ TVM_REGISTER_CUDA_TAG("nvidia/nvs-5200m", "sm_21", 49152, 32768); TVM_REGISTER_CUDA_TAG("nvidia/nvs-4200m", "sm_21", 49152, 32768); TVM_REGISTER_CUDA_TAG("nvidia/geforce-rtx-4090", "sm_89", 49152, 65536) .with_config("l2_cache_size_bytes", 75497472); +TVM_REGISTER_CUDA_TAG("nvidia/geforce-rtx-5060-ti", "sm_120", 49152, 65536) + .with_config("l2_cache_size_bytes", 33554432); TVM_REGISTER_CUDA_TAG("nvidia/geforce-rtx-3090-ti", "sm_86", 49152, 65536); TVM_REGISTER_CUDA_TAG("nvidia/geforce-rtx-3090", "sm_86", 49152, 65536); TVM_REGISTER_CUDA_TAG("nvidia/geforce-rtx-3080-ti", "sm_86", 49152, 65536);