py: Make MP_BC_SETUP_WITH use the bytecode stack for load_method.

dpgeorge · dpgeorge · commit c689c194710f · 2014-03-29T14:06:14.000Z
The compiler allocates 7 entries on the stack for a with statement
(following CPython, but probably can be reduced).  This is enough for
the method load and call in SETUP_WITH.
diff --git a/py/vm.c b/py/vm.c
@@ -381,16 +381,14 @@ mp_vm_return_kind_t mp_execute_byte_code_2(const byte *code_info, const byte **i
                         break;
                         */
 
-                    case MP_BC_SETUP_WITH: {
+                    case MP_BC_SETUP_WITH:
                         obj1 = TOP();
                         SET_TOP(rt_load_attr(obj1, MP_QSTR___exit__));
-                        mp_obj_t dest[2];
-                        rt_load_method(obj1, MP_QSTR___enter__, dest);
-                        obj2 = rt_call_method_n_kw(0, 0, dest);
+                        rt_load_method(obj1, MP_QSTR___enter__, sp + 1);
+                        obj2 = rt_call_method_n_kw(0, 0, sp + 1);
                         SETUP_BLOCK();
                         PUSH(obj2);
                         break;
-                    }
 
                     case MP_BC_WITH_CLEANUP: {
                         static const mp_obj_t no_exc[] = {mp_const_none, mp_const_none, mp_const_none};
