add bench check on CI

bytedance · Aug 11, 2024 · 5b5c7ee · 5b5c7ee
1 parent 7579b9c
commit 5b5c7ee
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/.github/workflows/benchmark-linux.yml b/.github/workflows/benchmark-linux.yml
@@ -27,4 +27,10 @@ jobs:
             ${{ runner.os }}-go-
 
       - name: Benchmark sonic
-        run: sh scripts/bench.sh
+        run: |
+          export SONIC_NO_ASYNC_GC=1
+          export SONIC_BENCH_SINGLE=1
+          ./scripts/bench.py -c -t 0.05 "go test -run ^$ -count=10 -benchmem -bench 'Benchmark(Encoder|Decoder)_(Generic|Binding)_Sonic' ./..."
+          ./scripts/bench.py -c -t 0.05 "go test -run ^$ -count=10 -benchmem -bench 'Benchmark(Get|Set)One_Sonic|BenchmarkParseSeven_Sonic' ./ast"
+          cd ./generic_test
+          ../scripts/bench.py -c -t 0.05 "go test -run ^$ -count=10 -benchmem -bench 'Benchmark' ."
diff --git a/generic_test/benchmark_test.go b/generic_test/benchmark_test.go
@@ -33,6 +33,7 @@ import (
 var (
     validFlag = os.Getenv("SONIC_VALID_GENERIC_BENCH")  != ""
     pretouchFlag = os.Getenv("SONIC_NO_PRETOUCH_BENCH") == ""
+    benchSingle = os.Getenv("SONIC_BENCH_SINGLE") != ""
 )
 
 type jsonLibEntry struct {
@@ -51,6 +52,15 @@ var jsonLibs = []jsonLibEntry {
     {"JsonIterStd", jsoniter.ConfigCompatibleWithStandardLibrary.Marshal, jsoniter.ConfigCompatibleWithStandardLibrary.Unmarshal},
 }
 
+func init() {
+    if benchSingle {
+        jsonLibs = []jsonLibEntry {
+            {"Sonic", sonic.Marshal, sonic.Unmarshal},
+            {"SonicStd", sonic.ConfigStd.Marshal, sonic.ConfigStd.Unmarshal},
+        }
+    }
+}
+
 func BenchmarkUnmarshalConcrete(b *testing.B)     { 
     runUnmarshalC(b)
 }