Update assignment expression logic for array subscripting

fruits-lab · Jul 7, 2024 · b3c4681 · b3c4681
1 parent dcff243
commit b3c4681
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/qbe_ir_generator.cpp b/src/qbe_ir_generator.cpp
@@ -1104,7 +1104,10 @@ void QbeIrGenerator::Visit(const SimpleAssignmentExprNode& assign_expr) {
     // Global array subscripting will return the target address instead of the
     // address of `id`.
     if (assign_expr.lhs->is_global &&
-        !dynamic_cast<ArrSubExprNode*>(assign_expr.lhs.get())) {
+        dynamic_cast<ArrSubExprNode*>(assign_expr.lhs.get())) {
+      WriteInstr_("storew {}, {}", FuncScopeTemp{rhs_num},
+                  FuncScopeTemp{reg_num_to_id_num.at(lhs_num)});
+    } else if (assign_expr.lhs->is_global) {
       WriteInstr_("storew {}, {}", FuncScopeTemp{rhs_num},
                   user_defined::GlobalPointer{reg_num_to_id.at(lhs_num)});
     } else {