diff options
| author | Zach Koopmans <zkoopmans@google.com> | 2021-02-12 12:53:51 -0800 |
|---|---|---|
| committer | gVisor bot <gvisor-bot@google.com> | 2021-02-12 12:55:33 -0800 |
| commit | 58a0a66900b6f38c8eee638a7c249bcc0a1bd9be (patch) | |
| tree | b80e8a528f22c947c207023baaddefe6c2073e11 /test/benchmarks/ml | |
| parent | c99ad8d541670e7fec7480b97d6d258460f041e3 (diff) | |
Rename params for iperf and tensorflow.
Rename operation params in iperf and tensorflow to
match other similar benchmarks.
PiperOrigin-RevId: 357250304
Diffstat (limited to 'test/benchmarks/ml')
| -rw-r--r-- | test/benchmarks/ml/BUILD | 1 | ||||
| -rw-r--r-- | test/benchmarks/ml/tensorflow_test.go | 11 |
2 files changed, 11 insertions, 1 deletions
diff --git a/test/benchmarks/ml/BUILD b/test/benchmarks/ml/BUILD index 285ec35d9..3425b8dad 100644 --- a/test/benchmarks/ml/BUILD +++ b/test/benchmarks/ml/BUILD @@ -18,5 +18,6 @@ benchmark_test( deps = [ "//pkg/test/dockerutil", "//test/benchmarks/harness", + "//test/benchmarks/tools", ], ) diff --git a/test/benchmarks/ml/tensorflow_test.go b/test/benchmarks/ml/tensorflow_test.go index b0e0c4720..8fa75d778 100644 --- a/test/benchmarks/ml/tensorflow_test.go +++ b/test/benchmarks/ml/tensorflow_test.go @@ -20,6 +20,7 @@ import ( "gvisor.dev/gvisor/pkg/test/dockerutil" "gvisor.dev/gvisor/test/benchmarks/harness" + "gvisor.dev/gvisor/test/benchmarks/tools" ) // BenchmarkTensorflow runs workloads from a TensorFlow tutorial. @@ -43,7 +44,15 @@ func BenchmarkTensorflow(b *testing.B) { defer machine.CleanUp() for name, workload := range workloads { - b.Run(name, func(b *testing.B) { + runName, err := tools.ParametersToName(tools.Parameter{ + Name: "operation", + Value: name, + }) + if err != nil { + b.Fatalf("Faile to parse param: %v", err) + } + + b.Run(runName, func(b *testing.B) { ctx := context.Background() b.ResetTimer() |