def setUp(self): cuda.memory_pool.free_all_blocks() self.h = function_hooks.CupyMemoryProfileHook() self.f1 = functions.math.exponential.Exp() self.f2 = functions.activation.relu.ReLU() self.x = numpy.random.uniform(-0.1, 0.1, (3, 5)).astype(numpy.float32) x = cuda.to_gpu(self.x) with self.h: self.f1.apply((chainer.Variable(x),)) self.f1.apply((chainer.Variable(x),)) self.f2.apply((chainer.Variable(x),)) self.f2.apply((chainer.Variable(x),))
def setUp(self): cuda.memory_pool.free_all_blocks() self.h = function_hooks.CupyMemoryProfileHook() f1 = functions.exp f2 = functions.relu self.x = numpy.random.uniform(-0.1, 0.1, (3, 5)).astype(numpy.float32) x = cuda.to_gpu(self.x) with self.h: f1(chainer.Variable(x)) f1(chainer.Variable(x)) f2(chainer.Variable(x)) f2(chainer.Variable(x))
def setUp(self): self.h = function_hooks.CupyMemoryProfileHook() self.f = functions.math.exponential.Exp() self.f.add_hook(self.h) self.x = numpy.random.uniform(-0.1, 0.1, (3, 5)).astype(numpy.float32) self.gy = numpy.random.uniform(-0.1, 0.1, (3, 5)).astype(numpy.float32)
def setUp(self): self.h = function_hooks.CupyMemoryProfileHook() self.l = SimpleLink() self.x = numpy.random.uniform(-0.1, 0.1, (3, 5)).astype(numpy.float32) self.gy = numpy.random.uniform(-0.1, 0.1, (3, 5)).astype(numpy.float32)
def test_choose_unit(self): h = function_hooks.CupyMemoryProfileHook() self.assertEqual((self.denomi, self.unit), h._choose_unit(self.bytes))