Skip to content

Fix thinking checkpoint rebuild on length stop - #77

Merged
antirez merged 1 commit into
antirez:mainfrom
ivanfioravanti:codex/server-thinking-checkpoint-gate
May 11, 2026
Merged

Fix thinking checkpoint rebuild on length stop#77
antirez merged 1 commit into
antirez:mainfrom
ivanfioravanti:codex/server-thinking-checkpoint-gate

Conversation

@ivanfioravanti

Copy link
Copy Markdown
Contributor

Solving #76
Length-stopped chat/thinking requests were synchronously rebuilding the thinking checkpoint before returning the response. Skip canonicalization for incomplete thinking turns so client wall time reflects the sampled response.

Client results before the patch:

Context | Prompt TPS | Gen TPS | Gen Tokens | Total Time --------|------------|---------|------------|------------
0.5k | 230.0 | 23.6 | 128 | 7.3s
1k | 296.5 | 20.4 | 128 | 9.0s
2k | 366.3 | 12.4 | 128 | 15.7s
4k | 308.2 | 7.4 | 128 | 30.0s

Client results after the patch:

Context | Prompt TPS | Gen TPS | Gen Tokens | Total Time --------|------------|---------|------------|------------
0.5k | 239.1 | 38.8 | 128 | 5.2s
1k | 295.3 | 38.6 | 128 | 6.0s
2k | 357.3 | 33.2 | 128 | 9.4s
4k | 307.9 | 31.0 | 128 | 17.1s

Tested with:
make all ds4_test && make test

Length-stopped chat/thinking requests were synchronously rebuilding the
thinking checkpoint before returning the response. Skip canonicalization for
incomplete thinking turns so client wall time reflects the sampled response.

Client results before the patch:

Context | Prompt TPS | Gen TPS | Gen Tokens | Total Time
--------|------------|---------|------------|------------
   0.5k |      230.0 |    23.6 |        128 |       7.3s
     1k |      296.5 |    20.4 |        128 |       9.0s
     2k |      366.3 |    12.4 |        128 |      15.7s
     4k |      308.2 |     7.4 |        128 |      30.0s

Client results after the patch:

Context | Prompt TPS | Gen TPS | Gen Tokens | Total Time
--------|------------|---------|------------|------------
   0.5k |      239.1 |    38.8 |        128 |       5.2s
     1k |      295.3 |    38.6 |        128 |       6.0s
     2k |      357.3 |    33.2 |        128 |       9.4s
     4k |      307.9 |    31.0 |        128 |      17.1s

Tested with:
make all ds4_test && make test
@antirez
antirez merged commit cdf8040 into antirez:main May 11, 2026
@antirez

antirez commented May 11, 2026

Copy link
Copy Markdown
Owner

🦾

hexxyan pushed a commit to hexxyan/ds4 that referenced this pull request May 27, 2026
…ng-checkpoint-gate

Fix thinking checkpoint rebuild on length stop
@ivanfioravanti
ivanfioravanti deleted the codex/server-thinking-checkpoint-gate branch July 5, 2026 12:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants