Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

gh-112075: Accessing a single element should optimistically avoid locking #115109

Merged
merged 2 commits into from
Feb 21, 2024

Conversation

DinoV
Copy link
Contributor

@DinoV DinoV commented Feb 6, 2024

Makes accessing a single element thread safe.

Adds tracking for whether a dictionary is shared or not, but some of this gets duplicated in #115108.

Does not yet deal with all of the atomic assignments that need to happen to make this 100% right, that'll come in a separate PR.

@DinoV DinoV force-pushed the nogil_dict_get_threadsafe branch from 1266799 to 2f46a39 Compare February 6, 2024 22:49
@DinoV DinoV force-pushed the nogil_dict_get_threadsafe branch 8 times, most recently from ef7c6ea to 54095de Compare February 7, 2024 20:31
@DinoV DinoV requested a review from colesbury February 7, 2024 20:48
@DinoV DinoV marked this pull request as ready for review February 7, 2024 20:58
Objects/dictobject.c Outdated Show resolved Hide resolved
Objects/dictobject.c Outdated Show resolved Hide resolved
Objects/dictobject.c Outdated Show resolved Hide resolved
Objects/dictobject.c Outdated Show resolved Hide resolved
Objects/dictobject.c Outdated Show resolved Hide resolved
Objects/dictobject.c Outdated Show resolved Hide resolved
Comment on lines 1792 to 1794
set_keys(mp, new_keys_object(interp, log2_newsize, unicode));
if (mp->ma_keys == NULL) {
mp->ma_keys = oldkeys;
set_keys(mp, oldkeys);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't think the possible temporary assignment to NULL is thread-safe without the GIL. Let's assign the result of new_keys_object() to a temporary variable and check that before assigning it to mp->ma_keys.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

When I was looking at this function when you made a comment about it on #114741 I realized that the assignment of it to a new empty table isn't right at all! We really need to allocate the new table, copy everything over, and then we can publish it.

@@ -3591,11 +3942,18 @@ dict___contains___impl(PyDictObject *self, PyObject *key)
if (hash == -1)
return NULL;
}
#ifdef Py_GIL_DISABLED
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does this differ from PyDict_Contains()? Can we just call that?

Objects/dictobject.c Outdated Show resolved Hide resolved
@@ -4157,10 +4534,20 @@ _PyDict_Contains_KnownHash(PyObject *op, PyObject *key, Py_hash_t hash)
PyObject *value;
Py_ssize_t ix;

#ifdef Py_GIL_DISABLED
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This also looks like contains_known_hash

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Or does contains_known_hash look like this? 🤔

@DinoV DinoV force-pushed the nogil_dict_get_threadsafe branch from 54095de to af7ddb9 Compare February 16, 2024 00:12
Copy link
Contributor

@colesbury colesbury left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@DinoV DinoV force-pushed the nogil_dict_get_threadsafe branch from af7ddb9 to d608fb3 Compare February 21, 2024 00:44
@DinoV DinoV merged commit 5407146 into python:main Feb 21, 2024
33 checks passed
woodruffw pushed a commit to woodruffw-forks/cpython that referenced this pull request Mar 4, 2024
…id locking (python#115109)

Makes accessing a single element thread safe and typically lock free
diegorusso pushed a commit to diegorusso/cpython that referenced this pull request Apr 17, 2024
…id locking (python#115109)

Makes accessing a single element thread safe and typically lock free
@DinoV DinoV deleted the nogil_dict_get_threadsafe branch May 31, 2024 18:23
LukasWoodtli pushed a commit to LukasWoodtli/cpython that referenced this pull request Jan 22, 2025
…id locking (python#115109)

Makes accessing a single element thread safe and typically lock free
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants