diff options
| author | Damien George <damien.p.george@gmail.com> | 2017-01-17 15:38:32 +1100 |
|---|---|---|
| committer | Damien George <damien.p.george@gmail.com> | 2017-02-16 18:38:06 +1100 |
| commit | cb6300697c7ea7f3634375f348982fa49a6df24f (patch) | |
| tree | 303c643188d7188e2ad8e757b46919690749fe82 /py | |
| parent | 30b42dd72d4aa310e8c96ccbaf75150f8649c262 (diff) | |
| download | micropython-cb6300697c7ea7f3634375f348982fa49a6df24f.tar.gz micropython-cb6300697c7ea7f3634375f348982fa49a6df24f.zip | |
py/runtime: Optimise case of identity iterator so it doesn't alloc RAM.
Diffstat (limited to 'py')
| -rw-r--r-- | py/runtime.c | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/py/runtime.c b/py/runtime.c index b9d7b72dcf..b792eabe7c 100644 --- a/py/runtime.c +++ b/py/runtime.c @@ -1103,6 +1103,13 @@ void mp_store_attr(mp_obj_t base, qstr attr, mp_obj_t value) { mp_obj_t mp_getiter(mp_obj_t o_in, mp_obj_iter_buf_t *iter_buf) { assert(o_in); + mp_obj_type_t *type = mp_obj_get_type(o_in); + + // Check for native getiter which is the identity. We handle this case explicitly + // so we don't unnecessarily allocate any RAM for the iter_buf, which won't be used. + if (type->getiter == mp_identity_getiter) { + return o_in; + } // if caller did not provide a buffer then allocate one on the heap if (iter_buf == NULL) { @@ -1110,7 +1117,6 @@ mp_obj_t mp_getiter(mp_obj_t o_in, mp_obj_iter_buf_t *iter_buf) { } // check for native getiter (corresponds to __iter__) - mp_obj_type_t *type = mp_obj_get_type(o_in); if (type->getiter != NULL) { mp_obj_t iter = type->getiter(o_in, iter_buf); if (iter != MP_OBJ_NULL) { |