def testGetMemoryInfoCPU(self):
with self.assertRaisesRegex(ValueError,
'Allocator stats not available'):
config.get_memory_info('CPU:0')
with self.assertRaisesRegex(ValueError,
'Allocator stats not available'):
config.get_memory_usage('CPU:0')
def testGetMemoryInfoCPU(self):
if test_util.IsMklEnabled():
# TODO(gzmkl) work with Google team to address design issue in allocator.h
self.skipTest('MklCPUAllocator does not throw exception. So skip test.')
with self.assertRaisesRegex(ValueError, 'Allocator stats not available'):
config.get_memory_info('CPU:0')
with self.assertRaisesRegex(ValueError, 'Allocator stats not available'):
config.get_memory_usage('CPU:0')
def testGetMemoryInfoBasic(self):
device = array_ops.zeros([]).backing_device
info = config.get_memory_info(device)
self.assertGreater(info['current'], 0)
self.assertGreater(info['peak'], 0)
self.assertEqual(info.keys(), {'current', 'peak'})
self.assertEqual(config.get_memory_usage(device), info['current'])
def testGetMemoryInfoUnknownDevice(self):
with self.assertRaisesRegex(ValueError, 'No matching devices found'):
config.get_memory_info('unknown_device:0')
with self.assertRaisesRegex(ValueError, 'No matching devices found'):
config.get_memory_usage('unknown_device:0')
def testGetMemoryUsageAmbiguousDevice(self):
if len(config.list_physical_devices('GPU')) < 2:
self.skipTest('Need at least 2 GPUs')
with self.assertRaisesRegex(ValueError, 'Multiple devices'):
config.get_memory_usage('GPU')
def testGetMemoryInfoUnknownDevice(self):
with self.assertRaisesRegex(ValueError, 'Failed parsing device name'):
config.get_memory_info('unknown_device')
with self.assertRaisesRegex(ValueError, 'Failed parsing device name'):
config.get_memory_usage('unknown_device')
def testGetMemoryInfoCPU(self):
with self.assertRaisesRegex(ValueError, 'CPU does not support'):
config.get_memory_info('CPU:0')
with self.assertRaisesRegex(ValueError, 'CPU does not support'):
config.get_memory_usage('CPU:0')
def testGetMemoryUsageSubstring(self):
self.assertGreater(config.get_memory_usage('GPU:0'), 0)
def testGetMemoryUsage(self):
device = array_ops.zeros([]).backing_device
self.assertGreater(config.get_memory_usage(device), 0)
