From 2730e4b0bcba80799ddc10f52081927848540f30 Mon Sep 17 00:00:00 2001 From: Reed Wanderman-Milne Date: Wed, 15 Apr 2020 13:38:36 -0700 Subject: [PATCH] Fix crash if set_visible_devices() used with tf.keras.mixed_precision, part 2. I thought I fixed this in https://github.com/tensorflow/tensorflow/commit/10e5748ddfcf0c60e5ef0a90bb72a34bc55190ec, but didn't, since I still called list_local_devices(). Fixes https://github.com/tensorflow/tensorflow/issues/38516. PiperOrigin-RevId: 306711358 Change-Id: Ic07ff827c42bae925bd20fc360d51bbc94f457c5 --- .../mixed_precision/experimental/device_compatibility_check.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/keras/mixed_precision/experimental/device_compatibility_check.py b/tensorflow/python/keras/mixed_precision/experimental/device_compatibility_check.py index 9279c37bb52..0b28661d262 100644 --- a/tensorflow/python/keras/mixed_precision/experimental/device_compatibility_check.py +++ b/tensorflow/python/keras/mixed_precision/experimental/device_compatibility_check.py @@ -154,8 +154,8 @@ def log_device_compatibility_check(policy_name, skip_local): if not context.executing_eagerly() or _logged_compatibility_check: return _logged_compatibility_check = True - device_attr_list = device_lib.list_local_devices() if not skip_local: + device_attr_list = device_lib.list_local_devices() _log_device_compatibility_check(policy_name, device_attr_list) return