bpo-37017: PyObject_CallMethodObjArgs uses LOAD_METHOD optimization (GH-13516)

Update PyObject_CallMethodObjArgs and _PyObject_CallMethodIdObjArgs
to use _PyObject_GetMethod to avoid creating a bound method object
in many cases.

On a microbenchmark of PyObject_CallMethodObjArgs calling a method on
an interpreted Python class, this optimization resulted in a 1.7x
speedup.
This commit is contained in:
Michael J. Sullivan 2019-05-26 00:23:34 -07:00 committed by Inada Naoki
parent 7114c6504a
commit 47dd2f9fd8
2 changed files with 35 additions and 18 deletions

View File

@ -0,0 +1,4 @@
Update :c:func:`PyObject_CallMethodObjArgs` and ``_PyObject_CallMethodIdObjArgs``
to use ``_PyObject_GetMethod`` to avoid creating a bound method object in many
cases.
Patch by Michael J. Sullivan.

View File

@ -1159,7 +1159,7 @@ _PyObject_CallMethodId_SizeT(PyObject *obj, _Py_Identifier *name,
/* --- Call with "..." arguments ---------------------------------- */
static PyObject *
object_vacall(PyObject *callable, va_list vargs)
object_vacall(PyObject *base, PyObject *callable, va_list vargs)
{
PyObject *small_stack[_PY_FASTCALL_SMALL_STACK];
PyObject **stack;
@ -1174,7 +1174,7 @@ object_vacall(PyObject *callable, va_list vargs)
/* Count the number of arguments */
va_copy(countva, vargs);
nargs = 0;
nargs = base ? 1 : 0;
while (1) {
PyObject *arg = va_arg(countva, PyObject *);
if (arg == NULL) {
@ -1196,7 +1196,12 @@ object_vacall(PyObject *callable, va_list vargs)
}
}
for (i = 0; i < nargs; ++i) {
i = 0;
if (base) {
stack[i++] = base;
}
for (; i < nargs; ++i) {
stack[i] = va_arg(vargs, PyObject *);
}
@ -1210,23 +1215,26 @@ object_vacall(PyObject *callable, va_list vargs)
}
PyObject *
PyObject_CallMethodObjArgs(PyObject *callable, PyObject *name, ...)
{
va_list vargs;
PyObject *result;
/* Private API for the LOAD_METHOD opcode. */
extern int _PyObject_GetMethod(PyObject *, PyObject *, PyObject **);
if (callable == NULL || name == NULL) {
PyObject *
PyObject_CallMethodObjArgs(PyObject *obj, PyObject *name, ...)
{
if (obj == NULL || name == NULL) {
return null_error();
}
callable = PyObject_GetAttr(callable, name);
PyObject *callable = NULL;
int is_method = _PyObject_GetMethod(obj, name, &callable);
if (callable == NULL) {
return NULL;
}
obj = is_method ? obj : NULL;
va_list vargs;
va_start(vargs, name);
result = object_vacall(callable, vargs);
PyObject *result = object_vacall(obj, callable, vargs);
va_end(vargs);
Py_DECREF(callable);
@ -1238,20 +1246,25 @@ PyObject *
_PyObject_CallMethodIdObjArgs(PyObject *obj,
struct _Py_Identifier *name, ...)
{
va_list vargs;
PyObject *callable, *result;
if (obj == NULL || name == NULL) {
return null_error();
}
callable = _PyObject_GetAttrId(obj, name);
if (callable == NULL) {
PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
if (!oname) {
return NULL;
}
PyObject *callable = NULL;
int is_method = _PyObject_GetMethod(obj, oname, &callable);
if (callable == NULL) {
return NULL;
}
obj = is_method ? obj : NULL;
va_list vargs;
va_start(vargs, name);
result = object_vacall(callable, vargs);
PyObject *result = object_vacall(obj, callable, vargs);
va_end(vargs);
Py_DECREF(callable);
@ -1266,7 +1279,7 @@ PyObject_CallFunctionObjArgs(PyObject *callable, ...)
PyObject *result;
va_start(vargs, callable);
result = object_vacall(callable, vargs);
result = object_vacall(NULL, callable, vargs);
va_end(vargs);
return result;