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

Fix a pytype error in system_info_lib.

Browse Source 
This allows us to turn on PY3 for t2t_trainer_xla.

PiperOrigin-RevId: 274979958
Change-Id: I386e05171b44ff93b8dc41943c1a3f7cead297d6
This commit is contained in:
Adrian Kuegel 2019-10-16 01:15:54 -07:00 committed by TensorFlower Gardener
parent 95bb950b57
commit c3d86dc087
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/system_info_lib.py
View File

@ -81,7 +81,7 @@ def gather_cpu_info():
# Gather num_cores_allowed
try:
with gfile.GFile('/proc/self/status', 'rb') as fh:
nc = re.search(r'(?m)^Cpus_allowed:\s*(.*)$', fh.read())
nc = re.search(r'(?m)^Cpus_allowed:\s*(.*)$', fh.read().decode('utf-8'))
if nc: # e.g. 'ff' => 8, 'fff' => 12
cpu_info.num_cores_allowed = (
bin(int(nc.group(1).replace(',', ''), 16)).count('1'))