Revert "[ROCm] Fix for ROCm CSB breakage - 200527"

This reverts commit 1c2527fd11.
2020-06-02 14:47:47 +00:00 · 2020-06-02 14:47:47 +00:00 · f38b02d59a
commit f38b02d59a
parent 4cb805dba2
1 changed files with 1 additions and 1 deletions
--- 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