Merge pull request lcompilers#1570 from anutosh491/branch08

virendrakabra14 · Mar 10, 2023 · ffdd221 · ffdd221
2 parents 4704837 + d88912c
commit ffdd221
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/integration_tests/CMakeLists.txt b/integration_tests/CMakeLists.txt
@@ -258,6 +258,7 @@ RUN(NAME loop_01             LABELS cpython llvm c)
 RUN(NAME loop_02             LABELS cpython llvm c wasm wasm_x86 wasm_x64)
 RUN(NAME loop_03             LABELS cpython llvm c wasm wasm_x64)
 RUN(NAME loop_04             LABELS cpython llvm)
+RUN(NAME loop_05             LABELS cpython llvm c)
 RUN(NAME if_01               LABELS cpython llvm c wasm wasm_x86 wasm_x64)
 RUN(NAME if_02               LABELS cpython llvm c wasm wasm_x86 wasm_x64)
 RUN(NAME print_02            LABELS cpython llvm c)

diff --git a/integration_tests/loop_05.py b/integration_tests/loop_05.py
@@ -0,0 +1,11 @@
+def f():
+    j :i32 = 0
+    while True:
+        j = j + 1
+        print(j)
+        if j > 4:
+            break
+    print("out of outer loop")
+    assert j == 5
+
+f()
diff --git a/src/libasr/pass/pass_utils.h b/src/libasr/pass/pass_utils.h
@@ -217,6 +217,12 @@ namespace LCompilers {
                     transform_stmts(xx.m_body, xx.n_body);
                 }
 
+                void visit_WhileLoop(const ASR::WhileLoop_t& x) {
+                    ASR::WhileLoop_t& xx = const_cast<ASR::WhileLoop_t&>(x);
+                    self().visit_expr(*xx.m_test);
+                    transform_stmts(xx.m_body, xx.n_body);
+                }
+
                 void visit_If(const ASR::If_t& x) {
                     ASR::If_t &xx = const_cast<ASR::If_t&>(x);
                     self().visit_expr(*xx.m_test);