only validate new_loop updated in replay

Author: jjsjann123

csrc/transform_replay.cpp

@@ -867,13 +867,20 @@ std::pair<TensorDomain*, int64_t> TransformReplay::replayCasP(
   }
 
   if (!opt.replay_allocation) {
+    // Duplicate TensorDomain to avoid validation run on existing TensorDomain.
     TensorDomain* replayed = IrBuilder::createInContainer<TensorDomain>(
         consumer->container(),
         consumer->getRootDomain(),
         consumer->getLogicalDomain(),
         consumer->getAllocationDomain(),
-        new_loop,
-        consumer->domain()->contiguity());
+        consumer->getLoopDomain(),
+        /*alternate_loop_domain=*/std::vector<IterDomain*>(),
+        consumer->domain()->contiguity(),
+        /*additiona_ids=*/std::vector<IterDomain*>(),
+        /*skip_validation=*/true);
+
+    // update loop domain, this ensures we run validation on new_loop.
+    replayed->setLoopDomain(new_loop);
 
     return {replayed, consumer_pos};
   }