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

Updating the regex match to catch p100 for run and gather logs."

Browse Source 
PiperOrigin-RevId: 162681705
This commit is contained in:
A. Unique TensorFlower 2017-07-20 16:25:58 -07:00 committed by TensorFlower Gardener
parent e4d6cb9744
commit 41ae3a8c0f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -135,7 +135,7 @@ def run_and_gather_logs(name, test_name, test_args,
gpu_config = gpu_info_lib.gather_gpu_devices() gpu_config = gpu_info_lib.gather_gpu_devices()
if gpu_config: if gpu_config:
gpu_name = gpu_config[0].model gpu_name = gpu_config[0].model
gpu_short_name_match = re.search(r"Tesla [KP][4,8]0", gpu_name) gpu_short_name_match = re.search(r"Tesla (K40|K80|P100)", gpu_name)
if gpu_short_name_match: if gpu_short_name_match:
gpu_short_name = gpu_short_name_match.group(0) gpu_short_name = gpu_short_name_match.group(0)
test_adjusted_name = name + "|" + gpu_short_name.replace(" ", "_") test_adjusted_name = name + "|" + gpu_short_name.replace(" ", "_")