Struct TensorOpScheduler 

pub struct TensorOpScheduler {
    pub ops: HashMap<u64, TensorOp>,
    /* private fields */
}

Priority-based tensor operation scheduler with dependency tracking.

See the module-level documentation for a full usage example.

Fields

ops: HashMap<u64, TensorOp>

All registered operations keyed by op_id.

Implementations

impl TensorOpScheduler

pub fn new() -> Self

Creates an empty scheduler.

pub fn enqueue( &mut self, name: String, priority: OpPriority, deps: Vec<u64>, flops: u64, ) -> u64

Enqueues a new operation and returns its assigned op_id.

The operation starts in OpStatus::Pending regardless of whether its dependency list is empty. Call advance_tick to promote operations with satisfied dependencies to OpStatus::Ready.

Arguments

• name – Human-readable operation name.
• priority – Scheduling priority.
• deps – IDs of operations that must complete before this one runs.
• flops – Estimated floating-point operations for resource accounting.

pub fn advance_tick(&mut self)

Advances the scheduler tick by one and re-evaluates dependency readiness.

Each OpStatus::Pending operation whose every dependency is in OpStatus::Completed is promoted to OpStatus::Ready.

pub fn start_op(&mut self, op_id: u64) -> bool

Attempts to transition the operation to OpStatus::Running.

Returns true on success. Returns false if the operation does not exist or is not in OpStatus::Ready state.

pub fn complete_op(&mut self, op_id: u64) -> bool

Attempts to transition the operation to OpStatus::Completed.

Returns true on success. Returns false if the operation does not exist or is not in OpStatus::Running state.

pub fn fail_op(&mut self, op_id: u64, reason: String) -> bool

Attempts to transition the operation to OpStatus::Failed.

The operation must be in OpStatus::Running or OpStatus::Ready state. Returns true on success, false otherwise.

pub fn next_ready(&self) -> Option<u64>

Returns the op_id of the highest-priority OpStatus::Ready operation.

Ties in priority are broken by the lowest op_id (FIFO within priority). Returns None if no operations are currently in Ready state.

pub fn stats(&self) -> SchedulerStats

Returns a snapshot of aggregate scheduler statistics.

Trait Implementations

impl Default for TensorOpScheduler

fn default() -> Self

Returns the “default value” for a type.