diff options
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() |