aboutsummaryrefslogtreecommitdiff
path: root/py/objtype.h
diff options
context:
space:
mode:
authorDamien George <damien.p.george@gmail.com>2019-11-13 21:05:34 +1100
committerDamien George <damien.p.george@gmail.com>2019-12-27 12:34:22 +1100
commitaacd61893952af24049fd64f3735551ff9fc14ea (patch)
tree60d1dd8fab30cd4462bbe6bfa98783b4e913bb7e /py/objtype.h
parent11b4524b39482736b84de585cf683ff5c7ba4e9b (diff)
py/runtime: Don't allocate iter buf for user-defined types.
A user-defined type that defines __iter__ doesn't need any memory to be pre-allocated for its iterator (because it can't use such memory). So optimise for this case by not allocating the iter-buf.
Diffstat (limited to 'py/objtype.h')
-rw-r--r--py/objtype.h3
1 files changed, 3 insertions, 0 deletions
diff --git a/py/objtype.h b/py/objtype.h
index 3fc8c6e1b..2c613b904 100644
--- a/py/objtype.h
+++ b/py/objtype.h
@@ -51,4 +51,7 @@ mp_obj_t mp_obj_instance_call(mp_obj_t self_in, size_t n_args, size_t n_kw, cons
// this needs to be exposed for the above macros to work correctly
mp_obj_t mp_obj_instance_make_new(const mp_obj_type_t *self_in, size_t n_args, size_t n_kw, const mp_obj_t *args);
+// this needs to be exposed for mp_getiter
+mp_obj_t mp_obj_instance_getiter(mp_obj_t self_in, mp_obj_iter_buf_t *iter_buf);
+
#endif // MICROPY_INCLUDED_PY_OBJTYPE_H