Skip to main content

register_buffer

svod_tensor::tensor_registry

Function register_buffer 

Source 
pub fn register_buffer(uop_id: u64, tensor_id: u64, buffer: Arc<Buffer>)
Expand description

Register a buffer for an existing tensor.

Used by realize() to associate output buffers with tensors for schedule lookups. Stores buffer in both BUFFERS map and TensorEntry.

ยงArguments

  • uop_id - The UOp ID to index under (for lookups)
  • tensor_id - The tensor ID that owns this buffer
  • buffer - The materialized buffer