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