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

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

Browse Source 
This reverts commit 1c2527fd11.
This commit is contained in:
Deven Desai 2020-06-02 14:47:47 +00:00
parent 4cb805dba2
commit f38b02d59a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/integration_test/gradient_checkpoint_test.py
View File

@ -75,7 +75,7 @@ def _limit_gpu_memory():
if gpus: if gpus:
tf.config.experimental.set_virtual_device_configuration( tf.config.experimental.set_virtual_device_configuration(
gpus[0], gpus[0],
[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1152)]) [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])
return True return True
return False return False