Add micro benchmark for magic method '__index__'.

Author: fangerer

graalpython/benchmarks/src/micro/cpyext/c-list-iterating-obj.py

@@ -0,0 +1,162 @@
+from . import ccompile
+
+code = """
+#include "Python.h"
+
+typedef struct {
+    PyObject_HEAD;
+    int bState;
+    uint64_t cnt;
+} NativeMagicMethodsObject;
+
+
+PyObject* nmm_index(PyObject* self) {
+    return PyLong_FromLong((((NativeMagicMethodsObject*)self)->cnt)++);
+}
+
+int nmm_bool(PyObject* self) {
+    int curState = ((NativeMagicMethodsObject*)self)->bState;
+    ((NativeMagicMethodsObject*)self)->bState = !curState;
+    return curState;
+}
+
+PyObject* nmm_iter(PyObject* self) {
+    return NULL;
+}
+
+
+static PyNumberMethods NativeMagicMethods_number_methods = {
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    nmm_bool,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    nmm_index,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    nmm_index,
+
+    0,
+    0,
+
+};
+
+static struct PyMethodDef NativeMagicMethods_methods[] = {
+    {NULL, NULL, 0, NULL},
+    {NULL, NULL, 0, NULL}
+};
+
+static PyTypeObject NativeMagicMethodsType = {
+    PyVarObject_HEAD_INIT(NULL, 0)
+        "NativeMagicMethods.NativeMagicMethods",
+    sizeof(NativeMagicMethodsObject),       /* tp_basicsize */
+    0,                          /* tp_itemsize */
+    0,                          /* tp_dealloc */
+    0,
+    0,
+    0,
+    0,                          /* tp_reserved */
+    0,
+    &NativeMagicMethods_number_methods,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    Py_TPFLAGS_DEFAULT,
+    "",
+    0,              /* tp_traverse */
+    0,                 /* tp_clear */
+    0,           /* tp_richcompare */
+    0,                          /* tp_weaklistoffset */
+    nmm_iter,                  /* tp_iter */
+    0,              /* tp_iternext */
+    NativeMagicMethods_methods,             /* tp_methods */
+    NULL,                       /* tp_members */
+    0,                          /* tp_getset */
+    0,                          /* tp_base */
+    0,                  /* tp_dict */
+    0,                          /* tp_descr_get */
+    0,                          /* tp_descr_set */
+    0,                          /* tp_dictoffset */
+    0,                  /* tp_init */
+    PyType_GenericAlloc,        /* tp_alloc */
+    PyType_GenericNew,          /* tp_new */
+    PyObject_Del,               /* tp_free */
+};
+
+static PyModuleDef NativeMagicMethodsmodule = {
+    PyModuleDef_HEAD_INIT,
+    "c_list_iterating_obj_module",
+    "",
+    -1,
+    NULL, NULL, NULL, NULL, NULL
+};
+
+PyMODINIT_FUNC
+PyInit_c_list_iterating_obj_module(void)
+{
+    PyObject* m;
+
+
+    if (PyType_Ready(&NativeMagicMethodsType) < 0)
+        return NULL;
+
+
+    m = PyModule_Create(&NativeMagicMethodsmodule);
+    if (m == NULL)
+        return NULL;
+
+    Py_INCREF(&NativeMagicMethodsType);
+    PyModule_AddObject(m, "NativeMagicMethods", (PyObject *)&NativeMagicMethodsType);
+    return m;
+}
+
+"""
+
+
+ccompile("c_list_iterating_obj_module", code)
+from . import c_list_iterating_obj_module
+
+def iterate_list(ll, num):
+    idxObj = c_list_iterating_obj_module.NativeMagicMethods()
+    for t in range(num):
+        item = ll[idxObj]
+    return item
+
+
+def measure(num):
+    last_item = iterate_list(list(range(num)), num)
+    print("Last item ", last_item)
+
+
+def __benchmark__(num=1000000):
+    measure(num)

graalpython/benchmarks/src/micro/list-iterating-obj.py

@@ -0,0 +1,21 @@
+class Idx:
+    __cnt = 0
+    def __index__(self):
+        cur = self.__cnt
+        self.__cnt += 1
+        return cur
+
+def iterate_list(ll, num):
+    idxObj = Idx()
+    for t in range(num):
+        item = ll[idxObj]
+    return item
+
+
+def measure(num):
+    last_item = iterate_list(list(range(num)), num)
+    print("Last item ", last_item)
+
+
+def __benchmark__(num=1000000):
+    measure(num)

mx.graalpython/mx_graalpython_bench_param.py

@@ -82,6 +82,7 @@
     'list-indexing': ITER_15 + ['1000000'],
     'list-iterating-explicit': ITER_25 + ['1000000'],
     'list-iterating': ITER_25 + ['1000000'],
+    'list-iterating-obj': ITER_15 + ['10000000'],
     'math-sqrt': ITER_15 + ['500000000'],
     'object-allocate': ITER_10 + ['5000'],
     'object-layout-change': ITER_15 + ['1000000'],