skainet-compile-dag/sk.ainet.lang.graph/DefaultGraphExecutionContext

DefaultGraphExecutionContext

class DefaultGraphExecutionContext(val baseOps: TensorOps = VoidTensorOps(), val phase: Phase = Phase.EVAL, val tensorDataFactory: TensorDataFactory = DenseTensorDataFactory(), val hooks: ForwardHooks? = null, val memoryInfo: MemoryInfo = MemoryInfo.getEmptyInfo(), val executionStats: ExecutionStats = ExecutionStats(), val createTapeFactory: (GraphExecutionContext) -> ExecutionTape = { _ -> DefaultExecutionTape() }, val computeGraph: ComputeGraph? = null, baseSink: OpSink = NoOpSink) : GraphExecutionContext(source)

Constructors

DefaultGraphExecutionContext

constructor(baseOps: TensorOps = VoidTensorOps(), phase: Phase = Phase.EVAL, tensorDataFactory: TensorDataFactory = DenseTensorDataFactory(), hooks: ForwardHooks? = null, memoryInfo: MemoryInfo = MemoryInfo.getEmptyInfo(), executionStats: ExecutionStats = ExecutionStats(), createTapeFactory: (GraphExecutionContext) -> ExecutionTape = { _ -> DefaultExecutionTape() }, computeGraph: ComputeGraph? = null, baseSink: OpSink = NoOpSink)

Types

object Companion

Properties

open override val baseOps: TensorOps

val computeGraph: ComputeGraph?

Optional compute graph used by GraphSink presets.

createTapeFactory

open override val createTapeFactory: (GraphExecutionContext) -> ExecutionTape

open override val currentTape: ExecutionTape?

Current execution tape (null if not recording)

open override val executionStats: ExecutionStats

open override val hooks: ForwardHooks?

open override val memoryInfo: MemoryInfo

open override val observers: ExecutionObserverRegistry

open override val ops: KspTensorOps

open override val phase: Phase

val session: TraceSession

open override val tapeStack: TapeStack

Tape stack for nested execution contexts

tensorDataFactory

open override val tensorDataFactory: TensorDataFactory

Functions

open override fun backward(targets: List<Tensor<*, *>>, sources: List<Tensor<*, *>>)

fun clearSession()

Clear cached tensor references from the trace session. Call between training batches to prevent memory accumulation.

open override fun collectGarbage()

Force garbage collection

inline fun <R> record(block: DefaultGraphExecutionContext.() -> R): Pair<ExecutionTape?, R>

Convenience helper to record within a block and return the produced tape (and keep existing graph).

resetExecutionStats

open override fun resetExecutionStats()

Reset execution statistics

open override fun startRecording()

open override fun stopRecording(): ExecutionTape?

stopRecordingAndGet

fun ~~stopRecordingAndGet~~(): ExecutionTape?

Helper for internal use that returns the tape