Make sure IterDomain::resize generates non-symbolic IDs

csrc/ir/interface_nodes.h

@@ -614,7 +614,11 @@ class NVF_API TensorView : public Val {
   //! Resize an IterDomain by expanding both the left and right sides
   //! by given widths. The resulting IterDomain has an extent of
   //! (left_expansion + axis->extent() + right_expansion).
-  TensorView* resize(int64_t axis, Val* left_expansion, Val* right_expansion);
+  TensorView* resize(
+      int64_t axis,
+      Val* left_expansion,
+      Val* right_expansion,
+      std::optional<IterType> iter_type = std::nullopt);
 
   // WARNING: rFactor does not return this TensorView, ir returns a new
   //  tensorview consumed by this!

csrc/ir/internal_base_nodes.h

@@ -690,7 +690,11 @@ class TensorDomain : public Val {
       SwizzleMode swizzle_mode = SwizzleMode::Data);
 
   // Resize an axis by left_expansion and right_expansion
-  void resize(int64_t axis, Val* left_expansion, Val* right_expansion);
+  void resize(
+      int64_t axis,
+      Val* left_expansion,
+      Val* right_expansion,
+      std::optional<IterType> iter_type = std::nullopt);
 
   // Transform TensorView according to merge and split transformations
   TensorDomain* view(const AnalyzeViewResult& view_analysis);

csrc/ir/nodes.cpp

@@ -3697,13 +3697,19 @@ void TensorDomain::swizzle(
 void TensorDomain::resize(
     int64_t axis,
     Val* left_expansion,
-    Val* right_expansion) {
+    Val* right_expansion,
+    std::optional<IterType> iter_type) {
   NVF_ERROR(nDims() > 0, "Tried to do resize on a 0-dim domain");
   axis = wrapDim(axis);
 
   IterDomain* id = this->axis(axis);
 
-  auto resized_id = IterDomain::resize(id, left_expansion, right_expansion);
+  auto resized_id = IterDomain::resize(
+      id,
+      left_expansion,
+      right_expansion,
+      /*mark_as_rfactor=*/false,
+      iter_type);
   loop_domain_.at(axis) = resized_id;
   resetDomains();
 }

csrc/scheduler/tools/loop_domain_scheduler.cpp

@@ -151,10 +151,16 @@ class ReplayForwardTransformOnLoopDomain : OptInConstDispatch {
 
   void handle(const Resize* resize) final {
     NVF_ERROR(input_loop_ids_.size() == 1);
+    NVF_ERROR(
+        resize->out()->getIterType() != IterType::Symbolic,
+        "Unexpected to have a symbolic ID: ",
+        resize->out()->toString());
+    // Pass the iter type explicitly to avoid generating a symblic ID
     tv_->resize(
         getLoopIdPosition(input_loop_ids_.at(0)),
         resize->leftExpand(),
-        resize->rightExpand());
+        resize->rightExpand(),
+        resize->out()->getIterType());
   }
 
   void handle(const Swizzle2D* swizzle_2d) final {

csrc/tensor_view.cpp

@@ -527,7 +527,8 @@ TensorView* TensorView::merge(int64_t axis_o, int64_t axis_i) {
 TensorView* TensorView::resize(
     int64_t axis,
     Val* left_expansion,
-    Val* right_expansion) {
+    Val* right_expansion,
+    std::optional<IterType> iter_type) {
   NVF_ERROR(
       nDims() > 0,
       "Tried to do resize on a 0-dim TensorView. ",
@@ -560,7 +561,7 @@ TensorView* TensorView::resize(
       " Parallelization strategy must be set after calling resize: ",
       toString());
 
-  domain()->resize(axis, left_expansion, right_expansion);
+  domain()->resize(axis, left_expansion, right_expansion, iter_type);
   return this;
 }