Pointwise shouldn't check transpose scheduler (#3256)

Removed unnecessary check. Codegen diff kept failing with some python tests, which was due to #3283. Otherwise, no change is detected. --------- Co-authored-by: Jacob Hinkle <[email protected]>
NVIDIA · Oct 27, 2024 · 628a47e · 628a47e
1 parent 374fa38
commit 628a47e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 21 deletions.
diff --git a/csrc/scheduler/pointwise.cpp b/csrc/scheduler/pointwise.cpp
@@ -19,7 +19,6 @@
 #include <scheduler/registry_utils.h>
 #include <scheduler/runtime_info.h>
 #include <scheduler/tools/inlining.h>
-#include <scheduler/transpose.h>
 #include <scheduler/utils.h>
 #include <scheduler/vectorize_helper.h>
 
@@ -480,25 +479,6 @@ bool PointWiseScheduler::canScheduleCompileTime(Fusion* fusion) {
   return true;
 }
 
-bool PointWiseScheduler::canScheduleRunTime(
-    Fusion* fusion,
-    SchedulerRuntimeInfo& runtime_info,
-    HeuristicDataCache* data_cache) {
-  FUSER_PERF_SCOPE("PointWiseScheduler::canScheduleRunTime");
-  auto can_schedule_transpose_entry =
-      HeuristicDataCacheEntry<HeuristicCompileTime::CanScheduleTranspose>(
-          data_cache, [fusion]() {
-            return std::make_unique<bool>(
-                TransposeScheduler().canScheduleCompileTime(fusion));
-          });
-  if (can_schedule_transpose_entry.get()) {
-    return !TransposeScheduler().canScheduleRunTime(
-        fusion, runtime_info, data_cache);
-  }
-
-  return true;
-}
-
 // TODO: Inline intermediate operations (avoid inlining unrolled/vectorized
 // input/output caches)
 void schedulePointwise(Fusion* fusion, const PointwiseParams* pparams) {

diff --git a/csrc/scheduler/pointwise.h b/csrc/scheduler/pointwise.h
@@ -164,7 +164,9 @@ class PointWiseScheduler : public SchedulerEntry {
   bool canScheduleRunTime(
       Fusion* fusion,
       SchedulerRuntimeInfo& runtime_info,
-      HeuristicDataCache* data_cache = nullptr) override;
+      HeuristicDataCache* data_cache = nullptr) override {
+    return true;
+  }
 
   std::unique_ptr<HeuristicParams> computeHeuristics(
       Fusion* fusion,

diff --git a/tools/codediff/compare_codegen.sh b/tools/codediff/compare_codegen.sh
@@ -189,6 +189,7 @@ collect_kernels() {
     # Make tests reproducible
     export NVFUSER_TEST_RANDOM_SEED=0
     export NVFUSER_DISABLE=parallel_compile
+    export DEBUG_SERDE=true
     # run tests and benchmarks with cuda_to_file and dump output to files
 
     mkdir -p "$outdir/$commit"