skainet-compile-dag/sk.ainet.lang.graph.exec/GraphExecutionContext

GraphExecutionContext

interface GraphExecutionContext : ExecutionContext, TrainingExecutionContext(source)

Context for managing execution state, including mode switching, device management, and memory management.

Inheritors

DefaultGraphExecutionContext

Properties

abstract val baseOps: TensorOps

createTapeFactory

abstract val createTapeFactory: (executionContext: GraphExecutionContext) -> ExecutionTape

abstract val currentTape: ExecutionTape?

Current execution tape (null if not recording)

open val isRecording: Boolean

Whether operations should be recorded

abstract val tapeStack: TapeStack

Tape stack for nested execution contexts

Functions

abstract fun collectGarbage()

Force garbage collection

resetExecutionStats

abstract fun resetExecutionStats()

Reset execution statistics