From 58a0a66900b6f38c8eee638a7c249bcc0a1bd9be Mon Sep 17 00:00:00 2001
From: Zach Koopmans <zkoopmans@google.com>
Date: Fri, 12 Feb 2021 12:53:51 -0800
Subject: Rename params for iperf and tensorflow.

Rename operation params in iperf and tensorflow to
match other similar benchmarks.

PiperOrigin-RevId: 357250304
---
 test/benchmarks/ml/BUILD              |  1 +
 test/benchmarks/ml/tensorflow_test.go | 11 ++++++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

(limited to 'test/benchmarks/ml')

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()
-- 
cgit v1.2.3