当前位置:首页 > 科技  > 软件

如何在 GO 中写出准确的基准测试

来源: 责编: 时间:2023-11-28 09:36:41 151观看
导读一般来说,我们不应该对性能进行猜测。在编写优化时,会有许多因素可能起作用,即使我们对结果有很强的看法,测试它们很少是一个坏主意。然而,编写基准测试并不简单。很容易编写不准确的基准测试,并且基于这些测试得出错误的假

一般来说,我们不应该对性能进行猜测。在编写优化时,会有许多因素可能起作用,即使我们对结果有很强的看法,测试它们很少是一个坏主意。然而,编写基准测试并不简单。很容易编写不准确的基准测试,并且基于这些测试得出错误的假设。这篇文章的目标是探讨导致不准确的四个常见和具体陷阱:HIy28资讯网——每日最新资讯28at.com

  • 不重置或暂停计时器
  • 对微基准测试做出错误假设
  • 不注意编译器优化
  • 被观察效应所误导

HIy28资讯网——每日最新资讯28at.com

通用概念

在讨论这些陷阱之前,让我们简要回顾一下 Go 语言中基准测试的工作原理。基准测试的框架大致如下:HIy28资讯网——每日最新资讯28at.com

func BenchmarkFoo(b *testing.B) {    for i := 0; i < b.N; i++ {        foo()    }}

函数名以Benchmark前缀开头。被测试的函数(foo)在for循环内被调用。b.N代表着可变数量的迭代次数。在运行基准测试时,Go 会尝试使其匹配所请求的基准测试时间。基准测试时间默认设置为1秒,可以使用-benchtime标志进行更改。b.N从1开始;如果基准测试在1秒内完成,b.N会增加,然后再次运行基准测试,直到b.N大致匹配benchtime为止。HIy28资讯网——每日最新资讯28at.com

$ go test -bench=.cpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkFoo-4                73          16511228 ns/op

在这里,基准测试大约花费了1秒钟,foo被执行了73次,平均执行时间为16,511,228纳秒。我们可以使用-benchtime来更改基准测试时间:HIy28资讯网——每日最新资讯28at.com

$ go test -bench=. -benchtime=2sBenchmarkFoo-4               150          15832169 ns/op

foo 的执行次数大约是上一个基准测试的两倍。HIy28资讯网——每日最新资讯28at.com

接下来,让我们看一些常见的陷阱。HIy28资讯网——每日最新资讯28at.com

不重置或暂停计时器

在某些情况下,我们需要在基准测试循环之前执行一些操作。这些操作可能需要相当长的时间(例如,生成一个大型数据切片),可能会对基准测试结果产生显著影响:HIy28资讯网——每日最新资讯28at.com

func BenchmarkFoo(b *testing.B) {    expensiveSetup()    for i := 0; i < b.N; i++ {        functionUnderTest()    }}

在这种情况下,我们可以在进入循环之前使用ResetTimer方法:HIy28资讯网——每日最新资讯28at.com

func BenchmarkFoo(b *testing.B) {    expensiveSetup()    b.ResetTimer() // Reset the benchmark timer    for i := 0; i < b.N; i++ {        functionUnderTest()    }}

调用ResetTimer会将自测试开始以来的经过时间和内存分配计数器归零。这样,一个昂贵的设置步骤可以从测试结果中排除。HIy28资讯网——每日最新资讯28at.com

如果我们不仅需要在测试中执行一次昂贵的设置步骤,而是需要在每个循环迭代中都执行呢?HIy28资讯网——每日最新资讯28at.com

func BenchmarkFoo(b *testing.B) {    for i := 0; i < b.N; i++ {        expensiveSetup()        functionUnderTest()    }}

我们不能重置计时器,因为那样会在每次循环迭代期间执行。但是我们可以停止和恢复基准测试计时器,将调用expensiveSetup包裹起来:HIy28资讯网——每日最新资讯28at.com

func BenchmarkFoo(b *testing.B) {    for i := 0; i < b.N; i++ {        b.StopTimer() // Pause the benchmark timer        expensiveSetup()        b.StartTimer() // Resume the benchmark timer        functionUnderTest()    }}

在这里,我们暂停基准测试计时器来执行昂贵的设置步骤,然后恢复计时器。HIy28资讯网——每日最新资讯28at.com

注意: 关于这种方法有一个需要记住的地方:如果被测试的函数执行速度远远比设置函数要快,那么基准测试可能会花费太长时间才能完成。原因是它需要比 benchtime 更长的时间才能完成。基准测试时间的计算完全基于 functionUnderTest 的执行时间。所以,如果在每个循环迭代中等待了相当长的时间,基准测试就会比1秒要慢得多。如果我们想保持基准测试,一种可能的缓解方法是减少 benchtime 。HIy28资讯网——每日最新资讯28at.com

我们必须确保使用计时器方法来保持基准测试的准确性。HIy28资讯网——每日最新资讯28at.com

对微基准测试做出错误假设

微基准测试测量的是微小的计算单元,很容易对它做出错误的假设。比如说,我们不确定是应该使用atomic.StoreInt32还是atomic.StoreInt64(假设处理的值始终适合32位)。我们想编写一个基准测试来比较这两个函数:HIy28资讯网——每日最新资讯28at.com

func BenchmarkAtomicStoreInt32(b *testing.B) {    var v int32    for i := 0; i < b.N; i++ {        atomic.StoreInt32(&v, 1)    }}func BenchmarkAtomicStoreInt64(b *testing.B) {    var v int64    for i := 0; i < b.N; i++ {        atomic.StoreInt64(&v, 1)    }}

如果我们运行这个基准测试,以下是一些示例输出:HIy28资讯网——每日最新资讯28at.com

cpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkAtomicStoreInt32BenchmarkAtomicStoreInt32-4    197107742           5.682 ns/opBenchmarkAtomicStoreInt64BenchmarkAtomicStoreInt64-4    213917528           5.134 ns/op

我们很容易认为这个基准测试并且决定使用atomic.StoreInt64,因为它看起来更快。现在,为了进行一次公平的基准测试,我们改变顺序,先测试atomic.StoreInt64,然后测试atomic.StoreInt32。以下是一些示例输出:HIy28资讯网——每日最新资讯28at.com

BenchmarkAtomicStoreInt64BenchmarkAtomicStoreInt64-4    224900722           5.434 ns/opBenchmarkAtomicStoreInt32BenchmarkAtomicStoreInt32-4    230253900           5.159 ns/op

这次,atomic.StoreInt32有更好的结果。发生了什么事?HIy28资讯网——每日最新资讯28at.com

在微基准测试中,许多因素都会影响结果,比如在运行基准测试时的机器活动、电源管理、热管理和一系列指令的更好的缓存对齐。我们必须记住,许多因素,甚至超出我们 Go 项目范围的因素,都可能会影响结果。HIy28资讯网——每日最新资讯28at.com

注意: 我们应该确保运行基准测试的机器处于空闲状态。但是,外部进程可能在后台运行,这可能会影响基准测试结果。因此,像 perflock 这样的工具可以限制基准测试可以消耗的CPU。例如,我们可以使用总可用CPU的70%来运行基准测试,将30%分配给操作系统和其他进程,减少机器活动因素对结果的影响。HIy28资讯网——每日最新资讯28at.com

一个选择是使用-benchtime选项增加基准测试时间。类似于概率论中的大数定律,如果我们运行大量次数的基准测试,它应该趋向于接近其期望值(假设我们忽略了指令缓存等机制的好处)。HIy28资讯网——每日最新资讯28at.com

另一个选择是在经典基准测试工具之上使用外部工具。例如,benchstat工具是golang.org/x代码库的一部分,它允许我们计算和比较有关基准测试执行的统计数据。HIy28资讯网——每日最新资讯28at.com

让我们使用-count选项运行基准测试10次,并将输出重定向到一个特定的文件:HIy28资讯网——每日最新资讯28at.com

$ go test -bench=. -count=10 | tee stats.txtcpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkAtomicStoreInt32-4     234935682                5.124 ns/opBenchmarkAtomicStoreInt32-4     235307204                5.112 ns/op// ...BenchmarkAtomicStoreInt64-4     235548591                5.107 ns/opBenchmarkAtomicStoreInt64-4     235210292                5.090 ns/op// ...

然后我们可以在这个文件上运行benchstat:HIy28资讯网——每日最新资讯28at.com

$ benchstat stats.txtname                time/opAtomicStoreInt32-4  5.10ns ± 1%AtomicStoreInt64-4  5.10ns ± 1%

结果是一样的:两个函数的平均完成时间都是5.10纳秒。我们还可以看到给定基准测试执行间的百分比变化:±1%。这个指标告诉我们,两个基准测试都是稳定的,这让我们对计算出的平均结果更有信心。因此,我们不能得出atomic.StoreInt32比atomic.StoreInt64快或慢的结论,而是可以得出对于我们测试的用途(在特定的Go版本和特定的机器上),它们的执行时间是相似的。HIy28资讯网——每日最新资讯28at.com

总的来说,我们在进行微基准测试时应该保持谨慎。许多因素都可以对结果产生重大影响,并潜在地导致错误的假设。增加基准测试时间或重复基准测试的执行,并使用诸如benchstat之类的工具计算统计数据,可以是限制外部因素并获得更准确结果的有效方法,从而得出更好的结论。HIy28资讯网——每日最新资讯28at.com

我们还应该注意,如果另一个系统最终运行该应用程序,要小心使用在特定机器上执行的微基准测试的结果。生产系统的行为可能与我们运行微基准测试的系统大不相同。HIy28资讯网——每日最新资讯28at.com

不注意编译器优化

与编写基准测试相关的另一个常见错误是被编译器优化所欺骗,这也可能导致错误的基准测试假设。在这一节中,我们将看看Go语言的14813号问题(https://github.com/golang/go/issues/14813,也被Go项目成员Dave Cheney讨论过),涉及到一个人口统计函数(一个计算设置为1的位数的函数):HIy28资讯网——每日最新资讯28at.com

const m1 = 0x5555555555555555const m2 = 0x3333333333333333const m4 = 0x0f0f0f0f0f0f0f0fconst h01 = 0x0101010101010101func popcnt(x uint64) uint64 {    x -= (x >> 1) & m1    x = (x & m2) + ((x >> 2) & m2)    x = (x + (x >> 4)) & m4    return (x * h01) >> 56}

这个函数接受一个uint64并返回一个uint64。为了对这个函数进行基准测试,我们可以编写以下内容:HIy28资讯网——每日最新资讯28at.com

func BenchmarkPopcnt1(b *testing.B) {    for i := 0; i < b.N; i++ {        popcnt(uint64(i))    }}

然而,如果我们执行这个基准测试,结果会非常低得令人惊讶:HIy28资讯网——每日最新资讯28at.com

cpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkPopcnt1-4      1000000000               0.2858 ns/op

一个持续时间为0.28纳秒大约是一个时钟周期,所以这个数字是非常不合理地低。问题在于开发者对编译器优化不够谨慎。在这种情况下,被测试的函数非常简单,可以成为内联的候选对象:一种优化,用被调用函数的主体替换函数调用,让我们避免函数调用,这具有很小的开销。一旦函数被内联,编译器注意到调用没有副作用,并用以下基准测试替换了它:HIy28资讯网——每日最新资讯28at.com

func BenchmarkPopcnt1(b *testing.B) {    for i := 0; i < b.N; i++ {        // Empty    }}

现在基准测试是空的,这就是我们得到接近一个时钟周期结果的原因。为了防止这种情况发生,最佳实践是遵循以下模式:HIy28资讯网——每日最新资讯28at.com

  • 在每个循环迭代中,将结果赋值给一个局部变量(在基准测试函数的上下文中为局部变量)。
  • 将最新的结果赋值给一个全局变量。

在我们的情况下,我们编写以下基准测试:HIy28资讯网——每日最新资讯28at.com

var global uint64 // Define a global variablefunc BenchmarkPopcnt2(b *testing.B) {    var v uint64 // Define a local variable    for i := 0; i < b.N; i++ {        v = popcnt(uint64(i)) // Assign the result to the local variable    }    global = v // Assign the result to the global variable}

global 是一个全局变量,而 v 是一个局部变量,其作用域限定在基准测试函数内部。在每个循环迭代中,我们将 popcnt 的结果赋值给局部变量 v。然后将最新的结果赋值给全局变量 global。HIy28资讯网——每日最新资讯28at.com

如果我们运行这两个基准测试,现在结果之间会有显著的差异:HIy28资讯网——每日最新资讯28at.com

cpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkPopcnt1-4      1000000000               0.2858 ns/opBenchmarkPopcnt2-4      606402058                1.993 ns/op

BenchmarkPopcnt2 是基准测试的准确版本。它确保我们避免了内联优化,这种优化可能会人为降低执行时间,甚至去除对被测试函数的调用。依赖BenchmarkPopcnt1的结果可能会导致错误的假设。HIy28资讯网——每日最新资讯28at.com

让我们记住避免编译器优化误导基准测试结果的模式:将被测试函数的结果赋值给一个局部变量,然后将最新的结果赋值给一个全局变量。这种最佳实践还可以防止我们做出不正确的假设。HIy28资讯网——每日最新资讯28at.com

被观察效应的欺骗

在物理学中,观察者效应是通过观察行为干扰观察系统的现象。这种效应也可以在基准测试中看到,并可能导致对结果做出错误的假设。让我们看一个具体的例子,然后尝试减轻这种影响。HIy28资讯网——每日最新资讯28at.com

我们想要实现一个接收int64元素矩阵的函数。这个矩阵有固定的512列,我们想要计算前八列的总和,如图11.2所示。HIy28资讯网——每日最新资讯28at.com

HIy28资讯网——每日最新资讯28at.com

计算前八列的总和HIy28资讯网——每日最新资讯28at.com

为了优化的目的,我们还想确定变化列数是否会产生影响,所以我们也实现了一个有513列的第二个函数。实现如下:HIy28资讯网——每日最新资讯28at.com

func calculateSum512(s [][512]int64) int64 {    var sum int64    for i := 0; i < len(s); i++ { // Iterate over each row        for j := 0; j < 8; j++ { // Iterate over the first eight columns            sum += s[i][j] // Increment sum        }    }    return sum}func calculateSum513(s [][513]int64) int64 {    // Same implementation as calculateSum512}

我们遍历每一行,然后遍历前八列,并增加一个我们要返回的总和变量。在calculateSum513中的实现保持不变。HIy28资讯网——每日最新资讯28at.com

我们想对这些函数进行基准测试,以决定在固定行数的情况下哪一个性能最好:HIy28资讯网——每日最新资讯28at.com

const rows = 1000var res int64func BenchmarkCalculateSum512(b *testing.B) {    var sum int64    s := createMatrix512(rows) // Create a matrix of 512 columns    b.ResetTimer()    for i := 0; i < b.N; i++ {        sum = calculateSum512(s) // Create a matrix of 512 columns    }    res = sum}func BenchmarkCalculateSum513(b *testing.B) {    var sum int64    s := createMatrix513(rows) // Create a matrix of 513 columns    b.ResetTimer()    for i := 0; i < b.N; i++ {        sum = calculateSum513(s) // Calculate the sum    }    res = sum}

我们希望只创建一次矩阵,以限制对结果的影响。因此,我们在循环外调用createMatrix512和createMatrix513。我们可能期望结果会类似,因为我们只想遍历前八列,但事实并非如此(在我的机器上):HIy28资讯网——每日最新资讯28at.com

cpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkCalculateSum512-4        81854             15073 ns/opBenchmarkCalculateSum513-4       161479              7358 ns/op

第二个拥有513列的基准测试要快大约50%。再次强调,因为我们只遍历了前八列,这个结果相当令人惊讶。HIy28资讯网——每日最新资讯28at.com

要理解这种差异,我们需要了解CPU缓存的基础知识。简而言之,CPU由不同的缓存组成(通常是L1、L2和L3)。这些缓存降低了从主存储器访问数据的平均成本。在某些条件下,CPU可以从主存储器获取数据并将其复制到L1。在这种情况下,CPU试图将calculateSum感兴趣的矩阵子集(每行的前八列)放入L1。然而,一个情况下矩阵可以完全放入内存(513列),而另一个情况下不能(512列)。HIy28资讯网——每日最新资讯28at.com

回到基准测试,主要问题在于我们在两种情况下都重复使用相同的矩阵。因为该函数重复执行数千次,我们并未测量接收纯新矩阵的函数执行情况。相反,我们测量的是一个已经包含在缓存中的矩阵的子集的函数。因此,由于calculateSum513导致的缓存未命中更少,它具有更好的执行时间。HIy28资讯网——每日最新资讯28at.com

这是观察效应的一个例子。因为我们一直在观察重复调用的CPU密集型函数,CPU缓存可能会起作用,并且会对结果产生显著影响。在这个例子中,为了防止这种效应,我们应该在每次测试中创建一个新的矩阵,而不是重复使用一个:HIy28资讯网——每日最新资讯28at.com

func BenchmarkCalculateSum512(b *testing.B) {    var sum int64    for i := 0; i < b.N; i++ {        b.StopTimer()        s := createMatrix512(rows) // Create a new matrix during each loop iteration        b.StartTimer()        sum = calculateSum512(s)    }    res = sum}

现在在每次循环迭代中都创建了一个新的矩阵。如果我们再次运行基准测试(并调整benchtime——否则执行时间太长),结果会更接近:HIy28资讯网——每日最新资讯28at.com

cpu: Intel(R) Core(TM) i5-7360U CPU @ 2.30GHzBenchmarkCalculateSum512-4         1116             33547 ns/opBenchmarkCalculateSum513-4          998             35507 ns/op

不是做出calculateSum513更快的错误假设,我们看到当接收新矩阵时,两个基准测试的结果更接近。HIy28资讯网——每日最新资讯28at.com

正如我们在本文中看到的那样,因为我们重复使用了同一个矩阵,CPU缓存对结果产生了显著影响。为了防止这种情况发生,我们必须在每个循环迭代期间创建一个新的矩阵。总的来说,我们应该记住,在观察一个被测试函数的情况下,可能会在结果中产生显著差异,特别是在涉及低级优化的CPU密集型函数的微基准测试中。强制基准测试在每次迭代期间重新创建数据可能是防止这种影响的一种好方法。HIy28资讯网——每日最新资讯28at.com

结论

  • 使用时间方法来保持基准测试的准确性。
  • 当处理微基准测试时,增加benchtime或使用benchstat等工具可能会有所帮助。
  • 注意微基准测试的结果,如果最终运行应用程序的系统与运行微基准测试的系统不同。
  • 确保被测试函数产生了副作用,以防止编译器优化让您对基准测试结果产生误解。
  • 为了防止观察者效应,强制基准测试重新创建CPU密集型函数使用的数据。

本文链接:http://www.28at.com/showinfo-26-34662-0.html如何在 GO 中写出准确的基准测试

声明:本网页内容旨在传播知识,若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。邮件:2376512515@qq.com

上一篇: Prometheus 的查询语言 PromQL 详解

下一篇: 深入解析PyQt 6:为什么它比PyQt 5更强大?

标签:
  • 热门焦点
  • JavaScript 混淆及反混淆代码工具

    JavaScript 混淆及反混淆代码工具

    介绍在我们开始学习反混淆之前,我们首先要了解一下代码混淆。如果不了解代码是如何混淆的,我们可能无法成功对代码进行反混淆,尤其是使用自定义混淆器对其进行混淆时。什么是混
  • 把LangChain跑起来的三个方法

    把LangChain跑起来的三个方法

    使用LangChain开发LLM应用时,需要机器进行GLM部署,好多同学第一步就被劝退了,那么如何绕过这个步骤先学习LLM模型的应用,对Langchain进行快速上手?本片讲解3个把LangChain跑起来
  • 一年经验在二线城市面试后端的经验分享

    一年经验在二线城市面试后端的经验分享

    忠告这篇文章只适合2年内工作经验、甚至没有工作经验的朋友阅读。如果你是2年以上工作经验,请果断划走,对你没啥帮助~主人公这篇文章内容来自 「升职加薪」星球星友 的投稿,坐
  • 十个简单但很有用的Python装饰器

    十个简单但很有用的Python装饰器

    装饰器(Decorators)是Python中一种强大而灵活的功能,用于修改或增强函数或类的行为。装饰器本质上是一个函数,它接受另一个函数或类作为参数,并返回一个新的函数或类。它们通常用
  • 只需五步,使用start.spring.io快速入门Spring编程

    只需五步,使用start.spring.io快速入门Spring编程

    步骤1打开https://start.spring.io/,按照屏幕截图中的内容创建项目,添加 Spring Web 依赖项,并单击“生成”按钮下载 .zip 文件,为下一步做准备。请在进入步骤2之前进行解压。图
  • 网红炒股不为了赚钱,那就是耍流氓!

    网红炒股不为了赚钱,那就是耍流氓!

    来源:首席商业评论6月26日高调宣布入市,网络名嘴大v胡锡进居然进军了股市。在一次财经媒体峰会上,几个财经圈媒体大佬就&ldquo;胡锡进炒股是否知道认真报道&rdquo;展开讨论。有
  • 三星显示已开始为AR设备研发硅基LED微显示屏

    三星显示已开始为AR设备研发硅基LED微显示屏

    7月18日消息,据外媒报道,随着苹果首款头显产品Vision Pro在6月份正式推出,AR/VR/MR等头显产品也就将成为各大公司下一个重要的竞争领域,对显示屏这一关
  • 超级标准版旗舰!iQOO 11S全球首发iQOO超算独显芯片

    超级标准版旗舰!iQOO 11S全球首发iQOO超算独显芯片

    上半年已接近尾声,截至目前各大品牌旗下的顶级旗舰都已悉数亮相,而下半年即将推出的顶级旗舰已经成为了数码圈爆料的主流,其中就包括全新的iQOO 11S系
  • 联想YOGA 16s 2022笔记本将要推出,屏幕支持触控功能

    联想YOGA 16s 2022笔记本将要推出,屏幕支持触控功能

    联想此前宣布,将于11月2日19:30召开联想秋季轻薄新品发布会,推出联想 YOGA 16s 2022 笔记本等新品。官方称,YOGA 16s 2022 笔记本将搭载 16 英寸屏幕,并且是一
Top