diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-20 16:25:58 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-20 16:30:03 -0700 |
commit | 41ae3a8c0f64cb316a64df21a2ac789f3e5bb008 (patch) | |
tree | 9eaf9631a4c004148f46ada34d11b8e81778d3d8 /tensorflow/tools/test | |
parent | e4d6cb9744d94d9a84369f4b8afda6ce89f1d1d3 (diff) |
Updating the regex match to catch p100 for run and gather logs."
PiperOrigin-RevId: 162681705
Diffstat (limited to 'tensorflow/tools/test')
-rw-r--r-- | tensorflow/tools/test/run_and_gather_logs_lib.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/test/run_and_gather_logs_lib.py b/tensorflow/tools/test/run_and_gather_logs_lib.py index 570e09f165..c798dd5de7 100644 --- a/tensorflow/tools/test/run_and_gather_logs_lib.py +++ b/tensorflow/tools/test/run_and_gather_logs_lib.py @@ -135,7 +135,7 @@ def run_and_gather_logs(name, test_name, test_args, 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) + gpu_short_name_match = re.search(r"Tesla (K40|K80|P100)", gpu_name) if gpu_short_name_match: gpu_short_name = gpu_short_name_match.group(0) test_adjusted_name = name + "|" + gpu_short_name.replace(" ", "_") |