fix

NVIDIA · Dec 15, 2024 · 8056cfa · 8056cfa
1 parent 40dd2c2
commit 8056cfa
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/csrc/scheduler/resize.cpp b/csrc/scheduler/resize.cpp
@@ -73,10 +73,15 @@ bool ResizeScheduler::canScheduleCompileTime(Fusion* fusion) {
     return false;
   }
 
+  auto resize_out_tv =
+      resize_based_tensor_ops.at(0)->output(0)->as<TensorView>();
+
   auto all_dep_vals = DependencyCheck::getAllValsBetween(
-      {fusion->inputs().begin(), fusion->inputs().end()},
-      {resize_based_tensor_ops.at(0)->output(0)});
+      {fusion->inputs().begin(), fusion->inputs().end()}, {resize_out_tv});
   for (auto tv : ir_utils::filterByType<TensorView>(all_dep_vals)) {
+    if (tv == resize_out_tv) {
+      continue;
+    }
     if (tv->isFusionOutput()) {
       scheduler_debug_utils::canScheduleRejectReason(
           schedulerType(),