feature/benchmark/com.shifthackz.aisdv1.feature.benchmark/InferenceLikeWorkload

InferenceLikeWorkload

private class InferenceLikeWorkload(source)

Small deterministic workload shaped like diffusion inference stages.

It intentionally uses plain Kotlin arrays, bounded memory, and no model files, GPU contexts, delegates, or native runtime initialization.

Author

Dmitriy Moroz

Constructors

InferenceLikeWorkload

constructor()

Types

private object Companion

Properties

private val attention: FloatArray

private val bias: FloatArray

private val context: FloatArray

private var input: FloatArray

private val key: FloatArray

private var output: FloatArray

private val projection: FloatArray

private val query: FloatArray

private val tokens: FloatArray

private val value: FloatArray

private val weights: FloatArray

Functions

private fun attentionBlock(seed: Int): Float

private fun convolution3x3()

private fun decoderMix(seed: Int): Float

groupNormSiluResidual

private fun groupNormSiluResidual(block: Int)

private fun project(source: FloatArray, target: FloatArray, shift: Int)

fun runPass(seed: Int): Float

private fun sampleTokens(seed: Int)

private fun swapBuffers()