pub struct RingDeviceOutput {
pub output: Vec<Vec<f32>>,
pub lse: Vec<f32>,
pub transfers: usize,
}Expand description
Result from a single device in ring attention.
Fields§
§output: Vec<Vec<f32>>Output for this device’s Q shard.
lse: Vec<f32>LSE for this device’s Q shard.
transfers: usizeNumber of simulated ring transfers.
Trait Implementations§
Source§impl Clone for RingDeviceOutput
impl Clone for RingDeviceOutput
Source§fn clone(&self) -> RingDeviceOutput
fn clone(&self) -> RingDeviceOutput
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for RingDeviceOutput
impl RefUnwindSafe for RingDeviceOutput
impl Send for RingDeviceOutput
impl Sync for RingDeviceOutput
impl Unpin for RingDeviceOutput
impl UnsafeUnpin for RingDeviceOutput
impl UnwindSafe for RingDeviceOutput
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more