xref: /aosp_15_r20/external/ksp/benchmark/src/main/kotlin/com/google/devtools/ksp/BenchRunner.kt (revision af87fb4bb8e3042070d2a054e912924f599b22b7)
1 /*
2  * Copyright 2022 Google LLC
3  * Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 package com.google.devtools.ksp
18 
19 import org.jetbrains.kotlin.cli.jvm.K2JVMCompiler
20 import kotlin.system.measureNanoTime
21 
mainnull22 fun main(args: Array<String>) {
23     val runs = 3
24     val warmup = 3
25     val benchName = args[0]
26     val compilerArgs = args.drop(1).toTypedArray()
27     val cold = measureNanoTime{
28         K2JVMCompiler.main(compilerArgs)
29     } / 1000000L
30     for (i in 1..warmup) {
31         K2JVMCompiler.main(compilerArgs)
32     }
33     val hot = (1..runs).map { measureNanoTime{K2JVMCompiler.main(compilerArgs)} / 1000000L }
34     println("$benchName.Cold(RunTimeRaw): $cold ms")
35     println("$benchName.Hot(RunTimeRaw): ${hot.minOrNull()!!} ms")
36 }
37