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; }