-
-
Notifications
You must be signed in to change notification settings - Fork 32.2k
gh-119786: Add InternalDocs/qsbr.md. #135411
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
Draft
nascheme
wants to merge
3
commits into
python:main
Choose a base branch
from
nascheme:qsbr-internal-doc
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Draft
Changes from all commits
Commits
Show all changes
3 commits
Select commit
Hold shift + click to select a range
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change | ||||
---|---|---|---|---|---|---|
@@ -0,0 +1,129 @@ | ||||||
# Quiescent-State Based Reclamation (QSBR) | ||||||
|
||||||
## Introduction | ||||||
|
||||||
When implementing lock-free data structures, a key challenge is determining | ||||||
when it is safe to free memory that has been logically removed from a | ||||||
structure. Freeing memory too early can lead to use-after-free bugs if another | ||||||
thread is still accessing it. Freeing it too late results in excessive memory | ||||||
consumption. | ||||||
|
||||||
Safe memory reclamation (SMR) schemes address this by delaying the free | ||||||
operation until all concurrent read accesses are guaranteed to have completed. | ||||||
Quiescent-State Based Reclamation (QSBR) is a SMR scheme used in Python's | ||||||
free-threaded build to manage the lifecycle of shared memory. | ||||||
|
||||||
QSBR requires threads to periodically report that they are in a quiescent | ||||||
state. A thread is in a quiescent state if it holds no references to shared | ||||||
objects that might be reclaimed. Think of it as a checkpoint where a thread | ||||||
signals, "I am not in the middle of any operation that relies on a shared | ||||||
resource." In Python, the eval_breaker provides a natural and convenient place | ||||||
for threads to report this state. | ||||||
|
||||||
|
||||||
## Use in Free-Threaded Python | ||||||
|
||||||
While CPython's memory management is dominated by reference counting and a | ||||||
tracing garbage collector, these mechanisms are not suitable for all data | ||||||
structures. For example, the backing array of a list object is not individually | ||||||
reference-counted but may have a shorter lifetime than the PyListObject that | ||||||
contains it. We could delay reclamation until the next GC run, but we want | ||||||
reclamation to be prompt and to run the GC less frequently in the free-threaded | ||||||
build, as it requires pausing all threads. | ||||||
|
||||||
Many operations in the free-threaded build are protected by locks. However, for | ||||||
performance-critical code, we want to allow reads to happen concurrently with | ||||||
updates. For instance, we want to avoid locking during most list read accesses. | ||||||
If a list is resized while another thread is reading it, QSBR provides the | ||||||
mechanism to determine when it is safe to free the list's old backing array. | ||||||
|
||||||
Specific use cases for QSBR include: | ||||||
|
||||||
* Dictionary keys (PyDictKeysObject) and list arrays (_PyListArray): When a | ||||||
dictionary or list that may be shared between threads is resized, we use QSBR | ||||||
to delay freeing the old keys or array until it's safe. For dicts and lists | ||||||
that are not shared, their storage can be freed immediately upon resize. | ||||||
|
||||||
* Mimalloc mi_page_t: Non-locking dictionary and list accesses require | ||||||
cooperation from the memory allocator. If an object is freed and its memory is | ||||||
reused, we must ensure the new object's reference count field is at the same | ||||||
memory location. In practice, this means when a mimalloc page (mi_page_t) | ||||||
becomes empty, we don't immediately allow it to be reused for allocations of a | ||||||
different size class. QSBR is used to determine when it's safe to repurpose the | ||||||
page or return its memory to the OS. | ||||||
|
||||||
|
||||||
## Implementation Details | ||||||
|
||||||
|
||||||
### Core Implementation | ||||||
|
||||||
The proposal to add QSBR to Python is contained in Github issue 115103 [1]. | ||||||
Many details of that proposal have been copied here, so they can be kept | ||||||
up-to-date with the actual implementation. | ||||||
|
||||||
Python's QSBR implementation is based on FreeBSD's "Global Unbounded | ||||||
Sequences." [2, 3, 4]. It relies on a few key counters: | ||||||
|
||||||
* Global Write Sequence (`wr_seq`): A per-interpreter counter, `wr_seq`, is started | ||||||
at 1 and incremented by 2 each time it is advanced. This ensures its value is | ||||||
always odd, which can be used to distinguish it from other state values. When | ||||||
an object needs to be reclaimed, `wr_seq` is advanced, and the object is tagged | ||||||
with this new sequence number. | ||||||
|
||||||
* Per-Thread Read Sequence: Each thread has a local read sequence counter. When | ||||||
a thread reaches a quiescent state (e.g., at the eval_breaker), it copies the | ||||||
current global `wr_seq` to its local counter. | ||||||
|
||||||
* Global Read Sequence (`rd_seq`): This per-interpreter value stores the minimum | ||||||
of all per-thread read sequence counters (excluding detached threads). It is | ||||||
updated by a "polling" operation. | ||||||
|
||||||
To free an object, the following steps are taken: | ||||||
|
||||||
1. Advance the global `wr_seq`. | ||||||
|
||||||
2. Add the object's pointer to a deferred-free list, tagging it with the new | ||||||
`wr_seq` value as its qsbr_goal. | ||||||
|
||||||
Periodically, a polling mechanism processes this deferred-free list: | ||||||
|
||||||
1. The minimum read sequence value across all active threads is calculated and | ||||||
stored as the global `rd_seq`. | ||||||
|
||||||
2. For each item on the deferred-free list, if its qsbr_goal is less than the | ||||||
new `rd_seq`, its memory is freed, and it is removed from the list. Otherwise, | ||||||
it remains on the list for a future attempt. | ||||||
|
||||||
|
||||||
### Deferred Advance Optimization | ||||||
|
||||||
To reduce memory contention from frequent updates to the global `wr_seq`, its | ||||||
advancement is sometimes deferred. Instead of incrementing `wr_seq` on every | ||||||
reclamation request, each thread tracks its number of deferrals locally. Once | ||||||
the deferral count reaches a limit (QSBR_DEFERRED_LIMIT, currently 10), the | ||||||
thread advances the global `wr_seq` and resets its local count. | ||||||
|
||||||
When an object is added to the deferred-free list, its qsbr_goal is set to | ||||||
`wr_seq` + 2. By setting the goal to the next sequence value, we ensure it's safe | ||||||
to defer the global counter advancement. This optimization improves runtime | ||||||
speed but may increase peak memory usage by slightly delaying when memory can | ||||||
be reclaimed. | ||||||
|
||||||
|
||||||
## Limitations | ||||||
|
||||||
Determining the `rd_seq` requires scanning over all thread states. This operation | ||||||
could become a bottleneck in applications with a very large number of threads | ||||||
(e.g., >1,000). Future work may address this with more advanced mechanisms, | ||||||
such as a tree-based structure or incremental scanning. For now, the | ||||||
implementation prioritizes simplicity, with plans for refinement if | ||||||
multi-threaded benchmarks reveal performance issues. | ||||||
|
||||||
|
||||||
## References | ||||||
|
||||||
1. https://github.com/python/cpython/issues/115103 | ||||||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more.
Suggested change
Same for others |
||||||
2. https://youtu.be/ZXUIFj4nRjk?t=694 | ||||||
3. https://people.kernel.org/joelfernandes/gus-vs-rcu | ||||||
4. http://bxr.su/FreeBSD/sys/kern/subr_smr.c#44 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You can use github's footnotes for it