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
The following commit(s) were added to refs/heads/ci-docker-staging by this push:
new 370eca4d23 update min sm requirement
370eca4d23 is described below
commit 370eca4d239baff46a1df58e588b8f3e493c1149
Author: Masahiro Masuda <[email protected]>
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;
}