Reproduce an issue with MonitoredSession when saving a variable on a GPU.

Also arrange for continuous testing with GPUs.

PiperOrigin-RevId: 177895214

1 files changed, 14 insertions, 0 deletions

diff --git a/tensorflow/python/training/monitored_session_test.py b/tensorflow/python/training/monitored_session_test.py
index 159b2d5c16..349d8537cb 100644
--- a/tensorflow/python/training/monitored_session_test.py
+++ b/tensorflow/python/training/monitored_session_test.py
@@ -36,6 +36,7 @@ from tensorflow.python.framework import constant_op
 from tensorflow.python.framework import dtypes
 from tensorflow.python.framework import errors_impl
 from tensorflow.python.framework import ops
+from tensorflow.python.framework import test_util
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import control_flow_ops
 from tensorflow.python.ops import resource_variable_ops
@@ -1968,6 +1969,19 @@ class MonitoredSessionTest(test.TestCase):
           self.assertEqual(2, trace_the_exception['side_effect_counter'])
           self.assertNear(0.62, session.run(graph_state), 0.1)
 
+  def test_saver_on_a_gpu(self):
+    if not test_util.is_gpu_available():
+      return
+    with ops.Graph().as_default():
+      with self.test_session():
+        with ops.device('/gpu:0'):
+          variables.Variable(0)
+        saver_lib.Saver()
+
+        # TODO(b/36964652): Reproduces the issue that needs to be fixed.
+        with self.assertRaises(errors_impl.InvalidArgumentError):
+          monitored_session.MonitoredSession()
+
 
 class SingularMonitoredSessionTest(test.TestCase):
   """Tests SingularMonitoredSession."""