blob: 6e989a777ac7f01b0aa9ec18b020255f5e3bef81 [file] [log] [blame]
/*[clinic input]
preserve
[clinic start generated code]*/
#include "pycore_abstract.h" // _PyNumber_Index()
#include "pycore_modsupport.h" // _PyArg_CheckPositional()
PyDoc_STRVAR(_testinternalcapi_benchmark_locks__doc__,
"benchmark_locks($module, num_threads, work_inside=1, work_outside=0,\n"
" time_ms=1000, num_acquisitions=1, total_iters=0,\n"
" num_locks=1, random_locks=False, /)\n"
"--\n"
"\n");
#define _TESTINTERNALCAPI_BENCHMARK_LOCKS_METHODDEF \
{"benchmark_locks", _PyCFunction_CAST(_testinternalcapi_benchmark_locks), METH_FASTCALL, _testinternalcapi_benchmark_locks__doc__},
static PyObject *
_testinternalcapi_benchmark_locks_impl(PyObject *module,
Py_ssize_t num_threads,
int work_inside, int work_outside,
int time_ms, int num_acquisitions,
Py_ssize_t total_iters,
Py_ssize_t num_locks,
int random_locks);
static PyObject *
_testinternalcapi_benchmark_locks(PyObject *module, PyObject *const *args, Py_ssize_t nargs)
{
PyObject *return_value = NULL;
Py_ssize_t num_threads;
int work_inside = 1;
int work_outside = 0;
int time_ms = 1000;
int num_acquisitions = 1;
Py_ssize_t total_iters = 0;
Py_ssize_t num_locks = 1;
int random_locks = 0;
if (!_PyArg_CheckPositional("benchmark_locks", nargs, 1, 8)) {
goto exit;
}
{
Py_ssize_t ival = -1;
PyObject *iobj = _PyNumber_Index(args[0]);
if (iobj != NULL) {
ival = PyLong_AsSsize_t(iobj);
Py_DECREF(iobj);
}
if (ival == -1 && PyErr_Occurred()) {
goto exit;
}
num_threads = ival;
}
if (nargs < 2) {
goto skip_optional;
}
work_inside = PyLong_AsInt(args[1]);
if (work_inside == -1 && PyErr_Occurred()) {
goto exit;
}
if (nargs < 3) {
goto skip_optional;
}
work_outside = PyLong_AsInt(args[2]);
if (work_outside == -1 && PyErr_Occurred()) {
goto exit;
}
if (nargs < 4) {
goto skip_optional;
}
time_ms = PyLong_AsInt(args[3]);
if (time_ms == -1 && PyErr_Occurred()) {
goto exit;
}
if (nargs < 5) {
goto skip_optional;
}
num_acquisitions = PyLong_AsInt(args[4]);
if (num_acquisitions == -1 && PyErr_Occurred()) {
goto exit;
}
if (nargs < 6) {
goto skip_optional;
}
{
Py_ssize_t ival = -1;
PyObject *iobj = _PyNumber_Index(args[5]);
if (iobj != NULL) {
ival = PyLong_AsSsize_t(iobj);
Py_DECREF(iobj);
}
if (ival == -1 && PyErr_Occurred()) {
goto exit;
}
total_iters = ival;
}
if (nargs < 7) {
goto skip_optional;
}
{
Py_ssize_t ival = -1;
PyObject *iobj = _PyNumber_Index(args[6]);
if (iobj != NULL) {
ival = PyLong_AsSsize_t(iobj);
Py_DECREF(iobj);
}
if (ival == -1 && PyErr_Occurred()) {
goto exit;
}
num_locks = ival;
}
if (nargs < 8) {
goto skip_optional;
}
random_locks = PyObject_IsTrue(args[7]);
if (random_locks < 0) {
goto exit;
}
skip_optional:
return_value = _testinternalcapi_benchmark_locks_impl(module, num_threads, work_inside, work_outside, time_ms, num_acquisitions, total_iters, num_locks, random_locks);
exit:
return return_value;
}
/*[clinic end generated code: output=6cfed9fc081313ef input=a9049054013a1b77]*/