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

masahi pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit ac5cdadcbf5f0e2337a7ad9cb1518868c0830000
Author: Masahiro Masuda <masahi...@gmail.com>
AuthorDate: Tue Mar 21 17:05:48 2023 +0900

    update min sm requirement
---
 src/target/target_kind.cc | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
index a87bb92c48..d1b2c10edf 100644
--- a/src/target/target_kind.cc
+++ b/src/target/target_kind.cc
@@ -161,8 +161,8 @@ TargetJSON UpdateCUDAAttrs(TargetJSON target) {
     // Use the compute version of the first CUDA GPU instead
     TVMRetValue version;
     if (!DetectDeviceFlag({kDLCUDA, 0}, runtime::kComputeVersion, &version)) {
-      LOG(WARNING) << "Unable to detect CUDA version, default to 
\"-arch=sm_20\" instead";
-      archInt = 20;
+      LOG(WARNING) << "Unable to detect CUDA version, default to 
\"-arch=sm_50\" instead";
+      archInt = 50;
     } else {
       archInt = std::stod(version.operator std::string()) * 10 + 0.1;
     }
@@ -189,8 +189,8 @@ TargetJSON UpdateNVPTXAttrs(TargetJSON target) {
     // Use the compute version of the first CUDA GPU instead
     TVMRetValue version;
     if (!DetectDeviceFlag({kDLCUDA, 0}, runtime::kComputeVersion, &version)) {
-      LOG(WARNING) << "Unable to detect CUDA version, default to 
\"-mcpu=sm_20\" instead";
-      arch = 20;
+      LOG(WARNING) << "Unable to detect CUDA version, default to 
\"-mcpu=sm_50\" instead";
+      arch = 50;
     } else {
       arch = std::stod(version.operator std::string()) * 10 + 0.1;
     }

Reply via email to