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)