py/objgenerator: Save state in old_globals instead of local variable.

The code_state.old_globals variable is there to save the globals state so
should be used for this purpose, to avoid the need for additional local
variables on the C stack.
This commit is contained in:
Damien George 2018-05-22 16:39:19 +10:00
parent 20b4b85f72
commit 771cb359af

View File

@ -117,10 +117,10 @@ mp_vm_return_kind_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_
*self->code_state.sp = send_value; *self->code_state.sp = send_value;
} }
} }
mp_obj_dict_t *old_globals = mp_globals_get(); self->code_state.old_globals = mp_globals_get();
mp_globals_set(self->globals); mp_globals_set(self->globals);
mp_vm_return_kind_t ret_kind = mp_execute_bytecode(&self->code_state, throw_value); mp_vm_return_kind_t ret_kind = mp_execute_bytecode(&self->code_state, throw_value);
mp_globals_set(old_globals); mp_globals_set(self->code_state.old_globals);
switch (ret_kind) { switch (ret_kind) {
case MP_VM_RETURN_NORMAL: case MP_VM_RETURN_NORMAL: