Package-level declarations

Types

Link copied to clipboard
data class BenchmarkCase(val name: String, val description: String?, val warmupIterations: Int, val iterations: Int, val setup: BenchmarkExecutionScope.() -> Unit?, val execute: BenchmarkExecutionScope.() -> Unit, val teardown: BenchmarkExecutionScope.() -> Unit?)
Link copied to clipboard
Link copied to clipboard
data class BenchmarkCaseResult(val name: String, val description: String?, val warmupIterations: Int, val iterations: Int, val totalDuration: Duration, val durationPerIteration: Duration, val latency: LatencySummary?, val memory: MemorySummary?)
Link copied to clipboard
annotation class BenchmarkDsl
Link copied to clipboard
Link copied to clipboard
Link copied to clipboard
Link copied to clipboard
Link copied to clipboard
data class BenchmarkSuite(val name: String, val description: String?, val metrics: Set<BenchmarkMetric>, val contextFactory: () -> ExecutionContext, val cases: List<BenchmarkCase>)
Link copied to clipboard
Link copied to clipboard
data class BenchmarkSuiteResult(val suiteName: String, val description: String?, val metrics: Set<BenchmarkMetric>, val cases: List<BenchmarkCaseResult>)
Link copied to clipboard
Link copied to clipboard
data class LatencySummary(val measurements: List<LatencyMeasurement>, val totalDuration: Duration, val averageDuration: Duration, val count: Int, val totalsByOp: Map<String, Duration>)
Link copied to clipboard
data class MemorySummary(val samples: List<MemorySample>, val peakUsedBytes: Long, val lastUsedBytes: Long, val usagePercent: Double?)

Functions

Link copied to clipboard
Link copied to clipboard
fun <T : DType, V> runSlicingBenchmarks(config: SlicingBenchmarkConfig = SlicingBenchmarkConfig(), contextFactory: () -> ExecutionContext, tensorFactory: TensorFactory<T, V>): BenchmarkSuiteResult