Updating the regex match to catch p100 for run and gather logs."
PiperOrigin-RevId: 162681705
This commit is contained in:
parent
e4d6cb9744
commit
41ae3a8c0f
@ -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(" ", "_")
|
||||
|
Loading…
Reference in New Issue
Block a user