Skip to content

Commit 32b2d07

Browse files
Fidget-SpinnerFidget-Spinner
committed
Specialize arithmetic only on compact ints
1 parent afc5ab6 commit 32b2d07

File tree

11 files changed

+71
-44
lines changed

11 files changed

+71
-44
lines changed

‎Doc/howto/perf_profiling.rst‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ Then we can use ``perf report`` to analyze the data:
9292
| | | |
9393
| | | |--51.67%--_PyEval_EvalFrameDefault
9494
| | | | |
95-
| | | | |--11.52%--_PyLong_Add
95+
| | | | |--11.52%--_PyCompactLong_Add
9696
| | | | | |
9797
| | | | | |--2.97%--_PyObject_Malloc
9898
...
@@ -142,7 +142,7 @@ Instead, if we run the same experiment with ``perf`` support enabled we get:
142142
| | | |
143143
| | | |--51.81%--_PyEval_EvalFrameDefault
144144
| | | | |
145-
| | | | |--13.77%--_PyLong_Add
145+
| | | | |--13.77%--_PyCompactLong_Add
146146
| | | | | |
147147
| | | | | |--3.26%--_PyObject_Malloc
148148

‎Include/internal/pycore_long.h‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -112,9 +112,9 @@ PyAPI_DATA(PyObject*) _PyLong_Rshift(PyObject *, int64_t);
112112
// Export for 'math' shared extension
113113
PyAPI_DATA(PyObject*) _PyLong_Lshift(PyObject *, int64_t);
114114

115-
PyAPI_FUNC(PyObject*) _PyLong_Add(PyLongObject *left, PyLongObject *right);
116-
PyAPI_FUNC(PyObject*) _PyLong_Multiply(PyLongObject *left, PyLongObject *right);
117-
PyAPI_FUNC(PyObject*) _PyLong_Subtract(PyLongObject *left, PyLongObject *right);
115+
PyAPI_FUNC(PyObject*) _PyCompactLong_Add(PyLongObject *left, PyLongObject *right);
116+
PyAPI_FUNC(PyObject*) _PyCompactLong_Multiply(PyLongObject *left, PyLongObject *right);
117+
PyAPI_FUNC(PyObject*) _PyCompactLong_Subtract(PyLongObject *left, PyLongObject *right);
118118

119119
// Export for 'binascii' shared extension.
120120
PyAPI_DATA(unsigned char) _PyLong_DigitValue[256];

‎Include/internal/pycore_opcode_metadata.h‎

Lines changed: 3 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎Include/internal/pycore_uop_metadata.h‎

Lines changed: 3 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎Objects/longobject.c‎

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3772,9 +3772,11 @@ long_add(PyLongObject *a, PyLongObject *b)
37723772
}
37733773

37743774
PyObject *
3775-
_PyLong_Add(PyLongObject *a, PyLongObject *b)
3775+
_PyCompactLong_Add(PyLongObject *a, PyLongObject *b)
37763776
{
3777-
return (PyObject*)long_add(a, b);
3777+
assert(_PyLong_BothAreCompact(a, b));
3778+
stwodigits z = medium_value(a) + medium_value(b);
3779+
return (PyObject *)_PyLong_FromSTwoDigits(z);
37783780
}
37793781

37803782
static PyObject *
@@ -3815,9 +3817,10 @@ long_sub(PyLongObject *a, PyLongObject *b)
38153817
}
38163818

38173819
PyObject *
3818-
_PyLong_Subtract(PyLongObject *a, PyLongObject *b)
3820+
_PyCompactLong_Subtract(PyLongObject *a, PyLongObject *b)
38193821
{
3820-
return (PyObject*)long_sub(a, b);
3822+
assert(_PyLong_BothAreCompact(a, b));
3823+
return (PyObject *)_PyLong_FromSTwoDigits(medium_value(a) - medium_value(b));
38213824
}
38223825

38233826
static PyObject *
@@ -4262,9 +4265,11 @@ long_mul(PyLongObject *a, PyLongObject *b)
42624265
}
42634266

42644267
PyObject *
4265-
_PyLong_Multiply(PyLongObject *a, PyLongObject *b)
4268+
_PyCompactLong_Multiply(PyLongObject *a, PyLongObject *b)
42664269
{
4267-
return (PyObject*)long_mul(a, b);
4270+
assert(_PyLong_BothAreCompact(a, b));
4271+
stwodigits v = medium_value(a) * medium_value(b);
4272+
return (PyObject *)_PyLong_FromSTwoDigits(v);
42684273
}
42694274

42704275
static PyObject *

‎Python/bytecodes.c‎

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -582,9 +582,10 @@ dummy_func(
582582
PyObject *right_o = PyStackRef_AsPyObjectBorrow(right);
583583
assert(PyLong_CheckExact(left_o));
584584
assert(PyLong_CheckExact(right_o));
585+
DEOPT_IF(!_PyLong_BothAreCompact((PyLongObject *)left_o, (PyLongObject *)right_o));
585586

586587
STAT_INC(BINARY_OP, hit);
587-
PyObject *res_o = _PyLong_Multiply((PyLongObject *)left_o, (PyLongObject *)right_o);
588+
PyObject *res_o = _PyCompactLong_Multiply((PyLongObject *)left_o, (PyLongObject *)right_o);
588589
PyStackRef_CLOSE_SPECIALIZED(right, _PyLong_ExactDealloc);
589590
PyStackRef_CLOSE_SPECIALIZED(left, _PyLong_ExactDealloc);
590591
INPUTS_DEAD();
@@ -597,9 +598,10 @@ dummy_func(
597598
PyObject *right_o = PyStackRef_AsPyObjectBorrow(right);
598599
assert(PyLong_CheckExact(left_o));
599600
assert(PyLong_CheckExact(right_o));
601+
DEOPT_IF(!_PyLong_BothAreCompact((PyLongObject *)left_o, (PyLongObject *)right_o));
600602

601603
STAT_INC(BINARY_OP, hit);
602-
PyObject *res_o = _PyLong_Add((PyLongObject *)left_o, (PyLongObject *)right_o);
604+
PyObject *res_o = _PyCompactLong_Add((PyLongObject *)left_o, (PyLongObject *)right_o);
603605
PyStackRef_CLOSE_SPECIALIZED(right, _PyLong_ExactDealloc);
604606
PyStackRef_CLOSE_SPECIALIZED(left, _PyLong_ExactDealloc);
605607
INPUTS_DEAD();
@@ -612,9 +614,10 @@ dummy_func(
612614
PyObject *right_o = PyStackRef_AsPyObjectBorrow(right);
613615
assert(PyLong_CheckExact(left_o));
614616
assert(PyLong_CheckExact(right_o));
617+
DEOPT_IF(!_PyLong_BothAreCompact((PyLongObject *)left_o, (PyLongObject *)right_o));
615618

616619
STAT_INC(BINARY_OP, hit);
617-
PyObject *res_o = _PyLong_Subtract((PyLongObject *)left_o, (PyLongObject *)right_o);
620+
PyObject *res_o = _PyCompactLong_Subtract((PyLongObject *)left_o, (PyLongObject *)right_o);
618621
PyStackRef_CLOSE_SPECIALIZED(right, _PyLong_ExactDealloc);
619622
PyStackRef_CLOSE_SPECIALIZED(left, _PyLong_ExactDealloc);
620623
INPUTS_DEAD();

‎Python/executor_cases.c.h‎

Lines changed: 15 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎Python/generated_cases.c.h‎

Lines changed: 18 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎Python/optimizer_bytecodes.c‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ dummy_func(void) {
225225
if (sym_is_const(ctx, left) && sym_is_const(ctx, right)) {
226226
assert(PyLong_CheckExact(sym_get_const(ctx, left)));
227227
assert(PyLong_CheckExact(sym_get_const(ctx, right)));
228-
PyObject *temp = _PyLong_Add((PyLongObject *)sym_get_const(ctx, left),
228+
PyObject *temp = _PyCompactLong_Add((PyLongObject *)sym_get_const(ctx, left),
229229
(PyLongObject *)sym_get_const(ctx, right));
230230
if (temp == NULL) {
231231
goto error;
@@ -244,7 +244,7 @@ dummy_func(void) {
244244
if (sym_is_const(ctx, left) && sym_is_const(ctx, right)) {
245245
assert(PyLong_CheckExact(sym_get_const(ctx, left)));
246246
assert(PyLong_CheckExact(sym_get_const(ctx, right)));
247-
PyObject *temp = _PyLong_Subtract((PyLongObject *)sym_get_const(ctx, left),
247+
PyObject *temp = _PyCompactLong_Subtract((PyLongObject *)sym_get_const(ctx, left),
248248
(PyLongObject *)sym_get_const(ctx, right));
249249
if (temp == NULL) {
250250
goto error;
@@ -263,7 +263,7 @@ dummy_func(void) {
263263
if (sym_is_const(ctx, left) && sym_is_const(ctx, right)) {
264264
assert(PyLong_CheckExact(sym_get_const(ctx, left)));
265265
assert(PyLong_CheckExact(sym_get_const(ctx, right)));
266-
PyObject *temp = _PyLong_Multiply((PyLongObject *)sym_get_const(ctx, left),
266+
PyObject *temp = _PyCompactLong_Multiply((PyLongObject *)sym_get_const(ctx, left),
267267
(PyLongObject *)sym_get_const(ctx, right));
268268
if (temp == NULL) {
269269
goto error;

‎Python/optimizer_cases.c.h‎

Lines changed: 3 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)