(tvm) branch main updated: [TIR] Minor fix for default GPU schedule (#17706)

[syfeng]

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

syfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new e35a424262 [TIR] Minor fix for default GPU schedule (#17706)
e35a424262 is described below

commit e35a424262c2fcb8e0a70583c255657619f09f89
Author: Siyuan Feng <hzfen...@sjtu.edu.cn>
AuthorDate: Wed Mar 5 23:50:04 2025 +0800

    [TIR] Minor fix for default GPU schedule (#17706)
    
    Add checks for vulkan and opencl targets.
---
 src/tir/transforms/default_gpu_schedule.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tir/transforms/default_gpu_schedule.cc 
b/src/tir/transforms/default_gpu_schedule.cc
index ea521d6968..f61f385565 100644
--- a/src/tir/transforms/default_gpu_schedule.cc
+++ b/src/tir/transforms/default_gpu_schedule.cc
@@ -112,7 +112,7 @@ bool IsScheduledOnGPU(const BaseFunc& func) {
   if (target.defined()) {
     int dev_type = target->GetTargetDeviceType();
     if (!(dev_type == kDLCUDA || dev_type == kDLMetal || dev_type == kDLROCM ||
-          dev_type == kDLWebGPU)) {
+          dev_type == kDLVulkan || dev_type == kDLOpenCL || dev_type == 
kDLWebGPU)) {
       return false;
     }
   }