diff --git a/csrc/device_lower/lower2device.cpp b/csrc/device_lower/lower2device.cpp
index 9d39040f4be..2ce917e6842 100644
--- a/csrc/device_lower/lower2device.cpp
+++ b/csrc/device_lower/lower2device.cpp
@@ -353,6 +353,7 @@ IdModelOptions getIdModelOptions(Fusion* fusion) {
       options.setConsumerIndex(true);
       options.setInlinePredicate(true);
       options.setUnswitchPredicate(true);
+      options.setLoop(true);
       continue;
     } else if (auto reshape = dynamic_cast<ViewOp*>(expr)) {
       // The legacy indexer has an issue when an expand broadcast is