Remove schduler_type from KernelExecutor.

NVIDIA · Jan 1, 2025 · c199020 · c199020
1 parent b458c7e
commit c199020
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/csrc/runtime/executor.cpp b/csrc/runtime/executor.cpp
@@ -176,7 +176,6 @@ void KernelExecutor::compile(
   NVF_ERROR(
       supported(fusion_.get()),
       "KernelExecutor does not support the Fusion provided.");
-  scheduler_type_ = scheduler_type;
 
   NVF_ERROR(
       !fusion_->outputs().empty(),
@@ -914,7 +913,7 @@ std::vector<at::Tensor> KernelExecutor::run(
         group_id_);
     SegmentProfiler& sprof = FusionProfiler::segment(group_id_);
     sprof.inputBytesAccessed(computeBytes(args));
-    sprof.scheduler(toString(scheduler_type_));
+    sprof.scheduler(toString(compiledKernel()->schedulerType()));
     FusionProfiler::segment(group_id_).setDevice(args.getDeviceIndex());
     sprof.startKernel();
   }
@@ -1197,7 +1196,7 @@ flatbuffers::Offset<serde::KernelExecutor> KernelExecutor::serialize(
       compiledKernel()->blockSizeHighWaterMark(),
       compiledKernel()->maxrregcountHighWaterMark(),
       warp_size_,
-      toUnderlying(scheduler_type_),
+      toUnderlying(compiledKernel()->schedulerType()),
       fusion_id_,
       concrete_id_,
       runtime_id_,
@@ -1367,8 +1366,6 @@ void KernelExecutor::deserialize(
       " vs ",
       buffer->heuristic());
 
-  scheduler_type_ = scheduler_type;
-
   auto device = c10::Device(c10::DeviceType::CUDA, device_index);
   c10::DeviceGuard dg(device);
 

diff --git a/csrc/runtime/executor.h b/csrc/runtime/executor.h
@@ -374,9 +374,6 @@ class KernelExecutor : public ExecutorAbstract {
   // Post-lowering hooks that are called to modify the kernel after lowering.
   // The main use case is for unit tests to modify the kernel.
   std::vector<std::function<void(kir::Kernel*)>> post_lowering_hooks_;
-
-  // TODO: Should this be removed?
-  SchedulerType scheduler_type_ = SchedulerType::None;
 };
 
 } // namespace nvfuser