oa::RolloutBuffer
RolloutBuffer class in the OA Ml public surface.
Public Methods
Static Public Methods
Constructor & Destructor Documentation
Public Method Documentation
Records one fused GPU append. The CPU cursor is control metadata only; no matrix data is read or copied through the host.
Parameters
inTransitionconst RolloutTransition &—
Returns
oa::StatusThe declared return value.
No public source comment is attached to this declaration.
Returns
const RolloutBatch &The declared return value.
No public source comment is attached to this declaration.
Returns
oa::U32The declared return value.
No public source comment is attached to this declaration.
Returns
const RolloutConfig &The declared return value.
Requires a complete rollout and records GAE directly into the preallocated advantage/return matrices.
Parameters
inConfigconst GaeConfig &Default: {}
Returns
oa::StatusThe declared return value.
No public source comment is attached to this declaration.
Returns
boolThe declared return value.
No public source comment is attached to this declaration.
Returns
boolThe declared return value.
No public source comment is attached to this declaration.
Returns
boolThe declared return value.
Begins a new collection cycle and clears valid on the GPU. Previously collected tensors remain allocated and are overwritten in place.
Returns
voidThe declared return value.
No public source comment is attached to this declaration.
Returns
oa::U32The declared return value.
No public source comment is attached to this declaration.
Parameters
inConfigconst RolloutConfig &—
Returns
oa::Result<RolloutBuffer>The declared return value.