From 140910c5415ac381459d8f31c4b9b4c18287964e Mon Sep 17 00:00:00 2001
From: Siyuan Feng <hzfengsy@sjtu.edu.cn>
Date: Wed, 5 Mar 2025 22:33:00 +0800
Subject: [PATCH] [TIR] Minor fix for default GPU schedule

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 ea521d696836..f61f385565cb 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;
     }
   }