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

Use test_adjusted_name when making the mangled_test_name in

Browse Source 
run_and_gather_logs_lib.py, to avoid duplicate file names when the same test is
run on multiple GPUs.

PiperOrigin-RevId: 157630193
This commit is contained in:
A. Unique TensorFlower 2017-05-31 13:54:32 -07:00 committed by TensorFlower Gardener
parent 0a84cfd588
commit a980aead87
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -131,8 +131,18 @@ def run_and_gather_logs(name, test_name, test_args,
# Hopefully running in sandboxed mode # Hopefully running in sandboxed mode
test_executable = os.path.join(".", test_executable) test_executable = os.path.join(".", test_executable)
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(" ", "_")
temp_directory = tempfile.mkdtemp(prefix="run_and_gather_logs") temp_directory = tempfile.mkdtemp(prefix="run_and_gather_logs")
mangled_test_name = name.strip("/").replace("/", "_").replace(":", "_") mangled_test_name = (test_adjusted_name.strip("/")
.replace("|", "_").replace("/", "_").replace(":", "_"))
test_file_prefix = os.path.join(temp_directory, mangled_test_name) test_file_prefix = os.path.join(temp_directory, mangled_test_name)
test_file_prefix = "%s." % test_file_prefix test_file_prefix = "%s." % test_file_prefix
@ -151,15 +161,6 @@ def run_and_gather_logs(name, test_name, test_args,
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(
test_adjusted_name, test_adjusted_name,
test_name=test_name, test_name=test_name,