You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2023/03/21 18:36:18 UTC
[tvm] 06/06: update min sm requirement
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 <ma...@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;
}