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

Merge pull request #35360 from ROCmSoftwarePlatform:google_upstream_rocm_platform_fix_191223

Browse Source 
PiperOrigin-RevId: 288038583
Change-Id: I10856b567692604377660439c7be662b443dfb2d
This commit is contained in:
TensorFlower Gardener 2020-01-03 13:23:05 -08:00
commit 9b32b2db11
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/python/framework/gpu_util.py
View File

@ -53,5 +53,5 @@ def compute_capability_from_device_desc(device_attrs):
# LINT.ThenChange(//tensorflow/core/common_runtime/gpu/gpu_device.cc)
if not match:
return GpuInfo(None, None)
cc = int(match.group(2)), int(match.group(3)) if match.group(2) else None
cc = (int(match.group(2)), int(match.group(3))) if match.group(2) else None
return GpuInfo(match.group(1), cc)