diff --git a/benchmark/softmax.cpp b/benchmark/softmax.cpp
index c7367c0f659..30551c0ecc9 100644
--- a/benchmark/softmax.cpp
+++ b/benchmark/softmax.cpp
@@ -98,9 +98,9 @@ static void Softmax_WarpReduceReference(benchmark::State& benchmark_state) {
   // Schedule through magic scheduler:
   SchedulerRuntimeInfo runtime_info(fusion, aten_inputs);
   NVF_ERROR(SchedulerEntry::canSchedule(
-      ScheduleHeuristic::InnerPersistent, fusion, runtime_info));
+      ScheduleHeuristic::Persistent, fusion, runtime_info));
   auto scheduler = SchedulerEntry::makeEntry(
-      ScheduleHeuristic::InnerPersistent, fusion, runtime_info);
+      ScheduleHeuristic::Persistent, fusion, runtime_info);
   scheduler->schedule(fusion);
 
   FusionExecutor fe;
diff --git a/csrc/scheduler/all_schedulers.h b/csrc/scheduler/all_schedulers.h
index 8109e7a32ec..3aced8c2f12 100644
--- a/csrc/scheduler/all_schedulers.h
+++ b/csrc/scheduler/all_schedulers.h
@@ -11,7 +11,6 @@
 #include <scheduler/normalization.h>
 #include <scheduler/normalization_inner.h>
 #include <scheduler/normalization_inner_outer.h>
-#include <scheduler/normalization_outer.h>
 #include <scheduler/pointwise.h>
 #include <scheduler/reduction.h>
 #include <scheduler/transpose.h>
diff --git a/csrc/scheduler/heuristic_types.cpp b/csrc/scheduler/heuristic_types.cpp
index c5d49ab4b02..f3ac55c075b 100644
--- a/csrc/scheduler/heuristic_types.cpp
+++ b/csrc/scheduler/heuristic_types.cpp
@@ -21,8 +21,6 @@ std::string toString(ScheduleHeuristic sh) {
       return "reduction";
     case ScheduleHeuristic::InnerPersistent:
       return "inner_persistent";
-    case ScheduleHeuristic::OuterPersistent:
-      return "outer_persistent";
     case ScheduleHeuristic::InnerOuterPersistent:
       return "inner_outer_persistent";
     case ScheduleHeuristic::Persistent:
diff --git a/csrc/scheduler/heuristic_types.h b/csrc/scheduler/heuristic_types.h
index 9ad831a204b..c634dd297ec 100644
--- a/csrc/scheduler/heuristic_types.h
+++ b/csrc/scheduler/heuristic_types.h
@@ -53,19 +53,17 @@ enum class ScheduleHeuristic {
   InnerPersistent,
   InnerOuterPersistent,
   Persistent,
-  OuterPersistent,
   Transpose,
   Matmul
 };
 
 //! Define a schedule table to loop over all the heuristics in priority order.
-constexpr std::array<ScheduleHeuristic, 9> all_heuristics_in_priority_order = {
+constexpr std::array<ScheduleHeuristic, 8> all_heuristics_in_priority_order = {
     ScheduleHeuristic::NoOp,
     ScheduleHeuristic::Reduction,
     ScheduleHeuristic::Transpose,
     ScheduleHeuristic::PointWise,
     ScheduleHeuristic::InnerPersistent,
-    ScheduleHeuristic::OuterPersistent,
     ScheduleHeuristic::InnerOuterPersistent,
     ScheduleHeuristic::Persistent,
     ScheduleHeuristic::Matmul};
diff --git a/csrc/scheduler/normalization.cpp b/csrc/scheduler/normalization.cpp
index ba4fc632aa4..6123b23c370 100644
--- a/csrc/scheduler/normalization.cpp
+++ b/csrc/scheduler/normalization.cpp
@@ -46,10 +46,6 @@ void PersistentKernelScheduler::schedule(Fusion* fusion) {
 }
 
 bool PersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
-  // This scheduler is being divided into three separate schedulers and should
-  // be deleted. Disable the use of this scheduler for now.
-  return false;
-
   // Needs at least one reduction to consider.
   auto reduction_ops = ir_utils::getReductionOps(fusion);
   if (reduction_ops.empty()) {
diff --git a/csrc/scheduler/normalization_outer.cpp b/csrc/scheduler/normalization_outer.cpp
index deb56f51291..0d15afb9409 100644
--- a/csrc/scheduler/normalization_outer.cpp
+++ b/csrc/scheduler/normalization_outer.cpp
@@ -36,7 +36,7 @@ OuterPersistentKernelScheduler::OuterPersistentKernelScheduler(
     Fusion* fusion,
     SchedulerRuntimeInfo& runtime_info,
     HeuristicSummary* data_cache)
-    : SchedulerEntry(ScheduleHeuristic::OuterPersistent) {
+    : SchedulerEntry(ScheduleHeuristic::Persistent) {
   computeHeuristics(fusion, runtime_info, data_cache);
 }
 
@@ -50,33 +50,33 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
   auto reduction_ops = ir_utils::getReductionOps(fusion);
   if (reduction_ops.empty()) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "needs a reduction op");
+        ScheduleHeuristic::Persistent, "needs a reduction op");
     return false;
   }
 
   if (ir_utils::filterByType<TensorView>(fusion->inputs()).empty()) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "Scheduling not supported with no input");
     return false;
   }
 
   // Check that inputs of all select/gather-like ops are fusion inputs
   if (registry_utils::rejectScheduleForMemoryPromotion(
-          fusion, ScheduleHeuristic::OuterPersistent)) {
+          fusion, ScheduleHeuristic::Persistent)) {
     return false;
   }
 
   // Fusions handled by persistent kernel scheduler cannot have MmaOp.
   if (!ir_utils::getMmaOps(fusion).empty()) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "no support for mma ops.");
+        ScheduleHeuristic::Persistent, "no support for mma ops.");
     return false;
   }
 
   if (registry_utils::hasNonUniqueBcast(fusion)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "Broadcasting dimension might be broadcasting to multiple sizes.");
     return false;
   }
@@ -86,7 +86,7 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
   if (reduction_tvs.empty()) {
     // Use pointwise logic
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "no reduction tv");
+        ScheduleHeuristic::Persistent, "no reduction tv");
     return false;
   }
 
@@ -101,14 +101,6 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
   }
   bool combined_inner_outer =
       !inner_reduction_tvs.empty() && !outer_reduction_tvs.empty();
-
-  if (!inner_reduction_tvs.empty() || outer_reduction_tvs.empty()) {
-    scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
-        "ScheduleHeuristic::OuterPersistent requires outer reduction tvs without inner reduction tvs.");
-    return false;
-  }
-
   if (!checkReductionPattern(
           fusion, inner_reduction_tvs, outer_reduction_tvs)) {
     return false;
@@ -123,7 +115,7 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
     ComputeAtMap ca_map(fusion);
     if (registry_utils::requiresForwardViewReplay(fusion, ca_map)) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent,
+          ScheduleHeuristic::Persistent,
           "Fusion requires view being reversible.");
       return false;
     }
@@ -133,7 +125,7 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
     if (registry_utils::reductionInterferingView(
             fusion, ca_map, reference_tv)) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent,
+          ScheduleHeuristic::Persistent,
           "View may interfere with normalization scheduling.");
       return false;
     }
@@ -161,7 +153,7 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
     } else {
       if (reduction_root_size(red) != axis_count) {
         scheduler_debug_utils::canScheduleRejectReason(
-            ScheduleHeuristic::OuterPersistent,
+            ScheduleHeuristic::Persistent,
             "inconsistent reduction root size: ",
             red->toString(),
             ", expected: ",
@@ -175,14 +167,14 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
   auto persistent_buffer_info = scheduler_utils::persistentBuffers(fusion);
   if (persistent_buffer_info.persistent_buffers.empty()) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "no persistent buffer identified");
+        ScheduleHeuristic::Persistent, "no persistent buffer identified");
     return false;
   }
 
   if (registry_utils::SchedulerTopologyChecker::
           hasNonNormalizePostReductionBCast(fusion)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "unsupported post reduction normalization");
     return false;
   }
@@ -190,7 +182,7 @@ bool OuterPersistentKernelScheduler::canScheduleCompileTime(Fusion* fusion) {
   if (registry_utils::SchedulerTopologyChecker::
           hasGatherToBroadcastBeforeReduction(fusion, reduction_tvs)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "has unsupported gather-like ops before normalization");
     return false;
   }
@@ -251,7 +243,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTime(
 
   if (persistent_buffer_size > available_persistent_buffer_size) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "not enough registers or shared memory for persistence");
     return false;
   }
@@ -277,7 +269,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTime(
                  false)
                  .first.has_value()) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent,
+          ScheduleHeuristic::Persistent,
           "Required batch number is larger than available batch number! Will cause register spills!");
       return false;
     }
@@ -300,8 +292,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTime(
   if (required_sm_per_norm >
       scheduler_utils::safeDiv(device_multiprocessor_count, 3)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
-        "requires over half GPU persistence.");
+        ScheduleHeuristic::Persistent, "requires over half GPU persistence.");
     return false;
   }
 
@@ -314,7 +305,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTime(
       !(norm_per_sm >= warp_size / 2 ||
         max_multi_reduction_factor >= warp_size)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "not enough threads");
+        ScheduleHeuristic::Persistent, "not enough threads");
     return false;
   }
 
@@ -330,7 +321,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTime(
                // half warp
                : (warp_size / 8) * device_multiprocessor_count)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "not enough blocks");
+        ScheduleHeuristic::Persistent, "not enough blocks");
     return false;
   }
 
@@ -360,7 +351,7 @@ bool OuterPersistentKernelScheduler::checkReductionPattern(
       if (!registry_utils::checkPatternEquivalence(
               rtvs[it - 1], rtvs[it], root_map)) {
         scheduler_debug_utils::canScheduleRejectReason(
-            ScheduleHeuristic::OuterPersistent,
+            ScheduleHeuristic::Persistent,
             "unmapped reduction ",
             rtvs[it - 1],
             " and ",
@@ -375,7 +366,7 @@ bool OuterPersistentKernelScheduler::checkReductionPattern(
     if (!normalization_scheduler_utils::checkIfReductionsAreInnerOuter(
             inner_reduction_tvs, outer_reduction_tvs)) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent,
+          ScheduleHeuristic::Persistent,
           "to use combined reduction, inner reduction tensor should be [I,I,...,R,R] and outer reduction tensor should be [R,R,...,I,I]");
       return false;
     }
@@ -383,7 +374,7 @@ bool OuterPersistentKernelScheduler::checkReductionPattern(
     if (!normalization_scheduler_utils::hasSharedInput(
             inner_reduction_tvs, outer_reduction_tvs)) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent,
+          ScheduleHeuristic::Persistent,
           "to use combined reduction, inner reduction and outer reduction should have shared input.");
       return false;
     }
@@ -391,7 +382,7 @@ bool OuterPersistentKernelScheduler::checkReductionPattern(
     if (!normalization_scheduler_utils::isConnectedOnlyThroughReductionProducer(
             inner_reduction_tvs, outer_reduction_tvs)) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent,
+          ScheduleHeuristic::Persistent,
           "to use combined reduction, inner reduction and outer reduction should not have shared consumer, their consumers should not have shared non-outer-reduction producer.");
       return false;
     }
@@ -529,8 +520,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
 
   if (persistent_buffer_size > available_persistent_buffer_size) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
-        "not enough registers for persistence");
+        ScheduleHeuristic::Persistent, "not enough registers for persistence");
     return false;
   }
 
@@ -557,7 +547,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
   if (required_sm_per_norm >
       scheduler_utils::safeDiv(device_multiprocessor_count, 2)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "requires over half GPU persistence.",
         " required SMs per normalization: ",
         required_sm_per_norm);
@@ -575,7 +565,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
     // TODO: Is this necessary for block persistence as well?
     if (vectorization_factor < 4) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent, "not enough vectorized");
+          ScheduleHeuristic::Persistent, "not enough vectorized");
       return false;
     }
 
@@ -589,7 +579,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
 
     if (!cross_grid_params.has_value()) {
       scheduler_debug_utils::canScheduleRejectReason(
-          ScheduleHeuristic::OuterPersistent, "no valid launch config found");
+          ScheduleHeuristic::Persistent, "no valid launch config found");
       return false;
     }
   }
@@ -606,7 +596,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
   // factor
   if (max_multi_reduction_factor < min_multi_reduction_factor) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "Not enough threads.",
         " Multi reduction factor, ",
         max_multi_reduction_factor,
@@ -631,7 +621,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
   if (is_cross_grid &&
       max_used_sms < scheduler_utils::safeDiv(device_multiprocessor_count, 2)) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent,
+        ScheduleHeuristic::Persistent,
         "cross grid - not enough used SMs: ",
         max_used_sms);
     return false;
@@ -645,7 +635,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
           device_max_threads_per_multiprocessor * 4 && // Large reduction dim
       max_used_sms < min_fraction_of_sms) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "not enough used SMs");
+        ScheduleHeuristic::Persistent, "not enough used SMs");
     return false;
   }
 
@@ -662,7 +652,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
             return !reduction_tv->definition()->isA<WelfordOp>();
           })) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "non-Welford not enabled yet");
+        ScheduleHeuristic::Persistent, "non-Welford not enabled yet");
     return false;
   }
 
@@ -679,7 +669,7 @@ bool OuterPersistentKernelScheduler::canScheduleRunTimeOuter(
        0) &&
       device_prop->major == 7) {
     scheduler_debug_utils::canScheduleRejectReason(
-        ScheduleHeuristic::OuterPersistent, "iteration not evenly divided");
+        ScheduleHeuristic::Persistent, "iteration not evenly divided");
     return false;
   }
 
diff --git a/csrc/scheduler/registry.cpp b/csrc/scheduler/registry.cpp
index 57ce3ffbbad..9d13cbd4c90 100644
--- a/csrc/scheduler/registry.cpp
+++ b/csrc/scheduler/registry.cpp
@@ -185,9 +185,6 @@ bool SchedulerEntry::canSchedule(
     case ScheduleHeuristic::InnerPersistent:
       return checkCanSchedule<InnerPersistentKernelScheduler>(
           fusion, runtime_info, data_cache);
-    case ScheduleHeuristic::OuterPersistent:
-      return checkCanSchedule<OuterPersistentKernelScheduler>(
-          fusion, runtime_info, data_cache);
     case ScheduleHeuristic::InnerOuterPersistent:
       return checkCanSchedule<InnerOuterPersistentKernelScheduler>(
           fusion, runtime_info, data_cache);
@@ -230,10 +227,6 @@ std::unique_ptr<SchedulerEntry> SchedulerEntry::makeEntry(
       scheduler_entry = std::make_unique<InnerPersistentKernelScheduler>(
           fusion, runtime_info, data_cache);
       break;
-    case ScheduleHeuristic::OuterPersistent:
-      scheduler_entry = std::make_unique<OuterPersistentKernelScheduler>(
-          fusion, runtime_info, data_cache);
-      break;
     case ScheduleHeuristic::InnerOuterPersistent:
       scheduler_entry = std::make_unique<InnerOuterPersistentKernelScheduler>(
           fusion, runtime_info, data_cache);
@@ -318,11 +311,6 @@ HeuristicSummary::HeuristicSummary(
       InnerPersistentKernelScheduler::canScheduleRunTime(
           fusion, runtime_info, this);
       break;
-    case ScheduleHeuristic::OuterPersistent:
-      getOuterPersistentHeuristics(fusion, runtime_info, this);
-      OuterPersistentKernelScheduler::canScheduleRunTime(
-          fusion, runtime_info, this);
-      break;
     case ScheduleHeuristic::InnerOuterPersistent:
       getInnerOuterPersistentHeuristics(fusion, runtime_info, this);
       InnerOuterPersistentKernelScheduler::canScheduleRunTime(
@@ -394,7 +382,6 @@ void HeuristicSummary::validate() const {
       break;
     }
     case ScheduleHeuristic::InnerPersistent:
-    case ScheduleHeuristic::OuterPersistent:
     case ScheduleHeuristic::InnerOuterPersistent: {
       NVF_ERROR(entry_type_map_.count(EntryType::REDUCTION_TVS));
       NVF_ERROR(
diff --git a/test/test_gpu2.cpp b/test/test_gpu2.cpp
index c76ecf02fac..3753630abf5 100644
--- a/test/test_gpu2.cpp
+++ b/test/test_gpu2.cpp
@@ -6059,10 +6059,9 @@ TEST_F(NVFuserTest, FusionZeroSizeTensorNormalization_CUDA) {
   at::Tensor input0 = at::randn({2, 4}, options);
   at::Tensor input1 = at::randn({0}, options);
 
-  auto reduction_params =
-      getOuterPersistentHeuristics(&fusion, {input0, input1});
+  auto reduction_params = getPersistentHeuristics(&fusion, {input0, input1});
   NVF_CHECK(reduction_params, "Reduction schedule was not generated!");
-  scheduleOuterPersistentKernel(&fusion, *reduction_params);
+  schedulePersistentKernel(&fusion, *reduction_params);
 
   auto lparams = reduction_params->lparams;
   FusionExecutor fe;
@@ -9229,9 +9228,9 @@ TEST_F(NVFuserTest, FusionTestWarpSoftMax_CUDA) {
   // Schedule through magic scheduler
   SchedulerRuntimeInfo runtime_info(&fusion, aten_inputs);
   NVF_CHECK(SchedulerEntry::canSchedule(
-      ScheduleHeuristic::InnerPersistent, &fusion, runtime_info));
+      ScheduleHeuristic::Persistent, &fusion, runtime_info));
   auto scheduler = SchedulerEntry::makeEntry(
-      ScheduleHeuristic::InnerPersistent, &fusion, runtime_info);
+      ScheduleHeuristic::Persistent, &fusion, runtime_info);
   scheduler->schedule(&fusion);
 
   // Modify the schedule to use warp reduction
diff --git a/test/test_gpu_outer_reduction.cpp b/test/test_gpu_outer_reduction.cpp
index 31c5e453fde..9057774d74b 100644
--- a/test/test_gpu_outer_reduction.cpp
+++ b/test/test_gpu_outer_reduction.cpp
@@ -1464,7 +1464,7 @@ void grid_persistent_reduction_outer_norm_like_scheduler(
     const auto& scheduler_entry =
         runtime->schedulerHeuristics()->heuristicsList().at(0);
     NVF_CHECK(
-        scheduler_entry->heuristic() == ScheduleHeuristic::OuterPersistent,
+        scheduler_entry->heuristic() == ScheduleHeuristic::Persistent,
         "Unexpected heuristic was chosen: ",
         scheduler_entry->heuristic());
 
@@ -1622,7 +1622,7 @@ void grid_persistent_welford_outer_norm_like_scheduler(
     const auto& scheduler_entry =
         runtime->schedulerHeuristics()->heuristicsList().at(0);
     NVF_CHECK(
-        scheduler_entry->heuristic() == ScheduleHeuristic::OuterPersistent,
+        scheduler_entry->heuristic() == ScheduleHeuristic::Persistent,
         "Unexpected heuristic was chosen: ",
         scheduler_entry->heuristic());
 
@@ -1801,7 +1801,7 @@ void grid_persistent_batchnorm_scheduler(
     const auto& scheduler_entry =
         runtime->schedulerHeuristics()->heuristicsList().at(0);
     NVF_CHECK(
-        scheduler_entry->heuristic() == ScheduleHeuristic::OuterPersistent,
+        scheduler_entry->heuristic() == ScheduleHeuristic::Persistent,
         "Unexpected heuristic was chosen: ",
         scheduler_entry->heuristic());
 
@@ -1938,7 +1938,7 @@ void grid_persistent_reduction_outer_norm_bwd_like_scheduler(
     const auto& scheduler_entry =
         runtime->schedulerHeuristics()->heuristicsList().at(0);
     NVF_CHECK(
-        scheduler_entry->heuristic() == ScheduleHeuristic::OuterPersistent,
+        scheduler_entry->heuristic() == ScheduleHeuristic::Persistent,
         "Unexpected heuristic was chosen: ",
         scheduler_entry->heuristic());
 
@@ -2128,7 +2128,7 @@ void grid_persistent_batchnorm_bwd_scheduler(
     const auto& scheduler_entry =
         runtime->schedulerHeuristics()->heuristicsList().at(0);
     NVF_CHECK(
-        scheduler_entry->heuristic() == ScheduleHeuristic::OuterPersistent,
+        scheduler_entry->heuristic() == ScheduleHeuristic::Persistent,
         "Unexpected heuristic was chosen: ",
         scheduler_entry->heuristic());