From a21e60ac9c487d8392a1e9109b83205403627e6f Mon Sep 17 00:00:00 2001 From: Naoya Maruyama Date: Thu, 6 Mar 2025 08:35:49 -0800 Subject: [PATCH] Don't try to schedule EmbeddingFwdOp --- csrc/scheduler/registry.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/csrc/scheduler/registry.cpp b/csrc/scheduler/registry.cpp index 2c002b7ecbe..c1f1b33720d 100644 --- a/csrc/scheduler/registry.cpp +++ b/csrc/scheduler/registry.cpp @@ -33,11 +33,11 @@ bool checkCanSchedule(Fusion* fusion, SchedulerType scheduler_type) { FusionGuard fg(fusion); - // Fusions with `SdpaFwdOp/SdpaBwdOp` are only accepted in `ExprEval` + // These ops are are only accepted in `ExprEval` // scheduler, all other schedulers should reject them. - if (ir_utils::hasOpsOfType(fusion)) { + if (ir_utils::hasOpsOfType(fusion)) { scheduler_debug_utils::canScheduleRejectReason( - scheduler_type, "SdpaOps are not supported."); + scheduler_type, "Has unsupported ops"); return false; }