You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by "masahi (via GitHub)" <gi...@apache.org> on 2023/03/22 02:14:35 UTC

[GitHub] [tvm] masahi commented on a diff in pull request #14363: [CI] Update GPU image for CUDA 11.7

masahi commented on code in PR #14363:
URL: https://github.com/apache/tvm/pull/14363#discussion_r1144161151


##########
src/target/target_kind.cc:
##########
@@ -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;

Review Comment:
   SM20 is no longer supported by CUDA 11.7, it's too old (more than 10 years ago gpu)



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org