pub struct IlpRegistry { /* private fields */ }Expand description
Registry for ILP tensor masks.
Implementations§
Source§impl IlpRegistry
impl IlpRegistry
Sourcepub fn insert_mask(
&mut self,
name: String,
hard: CudaBuffer,
soft: CudaBuffer,
schema_size: usize,
)
pub fn insert_mask( &mut self, name: String, hard: CudaBuffer, soft: CudaBuffer, schema_size: usize, )
Register a dense ILP mask (hard + soft weight buffers).
Sourcepub fn insert_mask_from_sparse(
&mut self,
name: String,
schema_size: usize,
active_ijk: &[(u32, u32, u32)],
active_soft: &[f32],
budget: usize,
) -> Result<(), XlogError>
pub fn insert_mask_from_sparse( &mut self, name: String, schema_size: usize, active_ijk: &[(u32, u32, u32)], active_soft: &[f32], budget: usize, ) -> Result<(), XlogError>
Insert a mask built from sparse candidate data.
Performs deterministic top-k ranking (desc soft value, then lower index) and stores the selected (i,j,k) entries directly — no dense buffer.
Sourcepub fn insert_selected_mask(
&mut self,
name: String,
schema_size: usize,
active_entries: &[(u32, u32, u32)],
)
pub fn insert_selected_mask( &mut self, name: String, schema_size: usize, active_entries: &[(u32, u32, u32)], )
Insert an already-selected sparse mask, preserving caller order exactly.
Sourcepub fn insert_selected_mask_device(
&mut self,
name: String,
schema_size: usize,
candidate_order: Vec<(u32, u32, u32)>,
active_flags: CudaBuffer,
selected_count: usize,
)
pub fn insert_selected_mask_device( &mut self, name: String, schema_size: usize, candidate_order: Vec<(u32, u32, u32)>, active_flags: CudaBuffer, selected_count: usize, )
Insert a device-resident sparse mask with active flags on GPU.
Sourcepub fn has_sparse_device_mask(&self) -> bool
pub fn has_sparse_device_mask(&self) -> bool
Returns true if any registered mask uses the sparse-device representation.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for IlpRegistry
impl RefUnwindSafe for IlpRegistry
impl Send for IlpRegistry
impl Sync for IlpRegistry
impl Unpin for IlpRegistry
impl UnsafeUnpin for IlpRegistry
impl UnwindSafe for IlpRegistry
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