experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

This is running all the benchmark tests on k20, k40, and k80. We had to allow

Browse Source 
all cpu-platforms for k80.

PiperOrigin-RevId: 155777715
This commit is contained in:
A. Unique TensorFlower 2017-05-11 11:59:06 -07:00 committed by TensorFlower Gardener
parent f2bbf4b9e7
commit 5ccf7d3223
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/tools/test/run_and_gather_logs_lib.py
View File

@ -19,6 +19,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import os import os
import re
import shlex import shlex
import subprocess import subprocess
import tempfile import tempfile
@ -26,6 +27,7 @@ import time
from tensorflow.core.util import test_log_pb2 from tensorflow.core.util import test_log_pb2
from tensorflow.python.platform import gfile from tensorflow.python.platform import gfile
from tensorflow.tools.test import gpu_info_lib
from tensorflow.tools.test import system_info_lib from tensorflow.tools.test import system_info_lib
@ -93,7 +95,8 @@ def process_benchmarks(log_files):
return benchmarks return benchmarks
def run_and_gather_logs(name, test_name, test_args, benchmark_type): def run_and_gather_logs(name, test_name, test_args,
benchmark_type):
"""Run the bazel test given by test_name. Gather and return the logs. """Run the bazel test given by test_name. Gather and return the logs.
Args: Args:
@ -148,8 +151,17 @@ def run_and_gather_logs(name, test_name, test_args, benchmark_type):
if not log_files: if not log_files:
raise MissingLogsError("No log files found at %s." % test_file_prefix) raise MissingLogsError("No log files found at %s." % test_file_prefix)
test_adjusted_name = name
gpu_config = gpu_info_lib.gather_gpu_devices()
if gpu_config:
gpu_name = gpu_config[0].model
gpu_short_name_match = re.search(r"Tesla [KP][4,8]0", gpu_name)
if gpu_short_name_match:
gpu_short_name = gpu_short_name_match.group(0)
test_adjusted_name = name + "|" + gpu_short_name.replace(" ", "_")
return (process_test_logs( return (process_test_logs(
name, test_adjusted_name,
test_name=test_name, test_name=test_name,
test_args=test_args, test_args=test_args,
benchmark_type=benchmark_type, benchmark_type=benchmark_type,