|
Mila
Deep Neural Network Library
|
Buffers BufSize tokens before forwarding a contiguous span to Sink. More...
Public Member Functions | |
| BufferedTokenStreamer (Sink &sink) | |
| void | finish () |
| void | flush () |
| void | operator() (int32_t tok) |
Buffers BufSize tokens before forwarding a contiguous span to Sink.
Reduces per-token overhead when Sink involves cross-thread posting or I/O. Satisfies TokenStreamer via operator()(int32_t). Call finish() after generation completes to flush any partial buffer and invoke Sink::on_finish() if present.
Sink is held by reference; the caller must ensure Sink outlives this object.
| BufSize | Tokens to buffer before each flush. Defaults to 8. |
| Sink | A TokenSink callable. |