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
class BenchmarkCaseBuilder
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
Link copied to clipboard
class BenchmarkExecutionScope
Link copied to clipboard
Link copied to clipboard
class BenchmarkMetricsBuilder
Link copied to clipboard
object BenchmarkRunner
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
class BenchmarkSuiteBuilder
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
object BenchmarkSuites
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