From f38b02d59a7b95dd5fb8b3fc711b3ec09735b171 Mon Sep 17 00:00:00 2001 From: Deven Desai Date: Tue, 2 Jun 2020 14:47:47 +0000 Subject: [PATCH] Revert "[ROCm] Fix for ROCm CSB breakage - 200527" This reverts commit 1c2527fd1134753431796831dcf225ce0846862d. --- .../python/keras/integration_test/gradient_checkpoint_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/keras/integration_test/gradient_checkpoint_test.py b/tensorflow/python/keras/integration_test/gradient_checkpoint_test.py index 100f3ca2022..9d9e0a062b3 100644 --- a/tensorflow/python/keras/integration_test/gradient_checkpoint_test.py +++ b/tensorflow/python/keras/integration_test/gradient_checkpoint_test.py @@ -75,7 +75,7 @@ def _limit_gpu_memory(): if gpus: tf.config.experimental.set_virtual_device_configuration( gpus[0], - [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1152)]) + [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)]) return True return False