Add num_gpus to gpu_pcie_bandwidth benchmark metadata

GoogleCloudPlatform · Dec 15, 2016 · 117fd56 · 117fd56
1 parent 9ed4e99
commit 117fd56
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/perfkitbenchmarker/linux_benchmarks/gpu_pcie_bandwidth_benchmark.py b/perfkitbenchmarker/linux_benchmarks/gpu_pcie_bandwidth_benchmark.py
@@ -230,6 +230,7 @@ def Run(benchmark_spec):
   raw_results = []
   metadata = {}
   metadata['num_iterations'] = num_iterations
+  metadata['num_gpus'] = cuda_toolkit_8.QueryNumberOfGpus(vm)
   metadata['memory_clock_MHz'] = FLAGS.gpu_pcie_bandwidth_clock_speeds[0]
   metadata['graphics_clock_MHz'] = FLAGS.gpu_pcie_bandwidth_clock_speeds[1]
   run_command = ('%s/extras/demo_suite/bandwidthTest --device=all'