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

Avoid RuntimeException if device placement is nop

Browse Source 
This benefits colab tutorials, allowing a block of code to be re-run
without throwing an exception.

PiperOrigin-RevId: 247064427
This commit is contained in:
Gaurav Jain 2019-05-07 12:02:59 -07:00 committed by TensorFlower Gardener
parent 86b2e49082
commit 49c8db3828
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/eager/context.py
View File

@ -1194,6 +1194,9 @@ class Context(object):
@log_device_placement.setter @log_device_placement.setter
def log_device_placement(self, enabled): def log_device_placement(self, enabled):
if self._log_device_placement == enabled:
return
if self._context_handle is not None: if self._context_handle is not None:
raise RuntimeError( raise RuntimeError(
"Device placement logging must be set at program startup") "Device placement logging must be set at program startup")

6
tensorflow/python/framework/config_test.py
View File

@ -198,8 +198,10 @@ class ConfigTest(test.TestCase, parameterized.TestCase):
with self.assertRaises(RuntimeError): with self.assertRaises(RuntimeError):
context.set_log_device_placement(True) context.set_log_device_placement(True)
with self.assertRaises(RuntimeError):
context.set_log_device_placement(False) # If the setting the device placement is a no-op, do not throw a runtime
# exception.
context.set_log_device_placement(False)
@test_util.run_gpu_only @test_util.run_gpu_only
@reset_eager @reset_eager