revert unused changes

Author: jjsjann123

csrc/scheduler/registry.cpp

@@ -58,6 +58,8 @@ bool checkCanSchedule(Fusion* fusion, SchedulerType scheduler_type) {
     return false;
   }
 
+  // TODO: check PreprocessGroupedMatmulInputSf's output is in global memory / fusion output
+
   // Fusions with `MatmulOp, LinearOp, MmaOp` can only be accepted by Matmul
   // scheduler.
   if (scheduler_type != SchedulerType::Matmul &&
@@ -72,6 +74,7 @@ bool checkCanSchedule(Fusion* fusion, SchedulerType scheduler_type) {
         scheduler_type, "Connected fusion graph check failed!");
     return false;
   }
+
   if (IterDomainGraph(fusion, /*allow_self_mapping=*/true).hasSelfMapping()) {
     scheduler_debug_utils::canScheduleRejectReason(
         scheduler_type, "Iter domain graph check failed!");

csrc/scheduler/tools/domain_map.cpp

@@ -107,15 +107,6 @@ getIndexedConsumerToProducerMap(Fusion* fusion, const ComputeAtMap& ca_map) {
   return indexed_id_map;
 }
 
-std::vector<IterDomain*> getSchedulingIds(TensorView* tv) {
-  // if (tv->isDefinitionType<PreprocessGroupedMatmulInputSf>() && 
-  //     (tv->definition()->as<PreprocessGroupedMatmulInputSf>()->inputOffsets() == tv ||
-  //     tv->definition()->as<PreprocessGroupedMatmulInputSf>()->outputOffsets() == tv)) {
-  //   return tv->getMaybeRootDomain();
-  // }
-  return tv->getLogicalDomain();
-}
-
 } // namespace
 
 DomainMap::DomainMap(Fusion* fusion) : fusion_(fusion), ca_map_(fusion) {
@@ -145,7 +136,7 @@ bool DomainMap::areAllInputIdsMappedTo(TensorView* input_tv, TensorView* tv)
   // Erase all input concrete IDs mapped to the output domain
   // Ignore unresolved broadcast dimensions
   eraseifInputMappedThroughRootDomainAndIndexing(
-      in_concrete_ids, getSchedulingIds(tv));
+      in_concrete_ids, tv->getMaybeLogicalDomain());
 
   return in_concrete_ids.empty();
 }
@@ -224,7 +215,7 @@ bool DomainMap::areAllTargetIdsCoveredBy(
   // it's safe for target_tv to have them.
   std::unordered_set<IterDomain*> covered_source_ids;
   for (IterDomain* source_id_ref :
-       get_source_iter_domains(getSchedulingIds(reference_tv))) {
+       get_source_iter_domains(reference_tv->getMaybeLogicalDomain())) {
     covered_source_ids.insert(source_id_ref);
   }
   // It's safe to have unmapped broadcast IterDomain. There're quite a few tests