def check_forward(self, xp):
link = MyModel()
if xp is cuda.cupy:
with testing.assert_warns(DeprecationWarning):
link = link.to_gpu()
hook = link_hooks.TimerHook()
with hook:
link(chainer.Variable(xp.array([[7, 5]], numpy.float32)))
link(chainer.Variable(xp.array([[8, 1]], numpy.float32)))
# call_history
hist = hook.call_history
assert len(hist) == 6
assert all(len(h) == 2 for h in hist)
names = [h[0] for h in hist]
times = [h[1] for h in hist]
assert names == [
'Linear', 'Linear', 'MyModel', 'Linear', 'Linear', 'MyModel'
]
assert times[0] + times[1] < times[2]
assert times[3] + times[4] < times[5]
# summary
summary = hook.summary()
assert sorted(summary.keys()) == ['Linear', 'MyModel']
assert summary['Linear']['occurrence'] == 4
numpy.testing.assert_allclose(
summary['Linear']['elapsed_time'],
times[0] + times[1] + times[3] + times[4])
assert summary['MyModel']['occurrence'] == 2
numpy.testing.assert_allclose(summary['MyModel']['elapsed_time'],
times[2] + times[5])
# print_report
s = six.StringIO()
hook.print_report(unit=self.unit, file=s)
report = s.getvalue()
assert len(report.splitlines()) == 3
assert re.search(r'Linear +[.0-9a-z]+ +4', report) is not None
assert re.search(r'MyModel +[.0-9a-z]+ +2', report) is not None
def test_name(self):
assert link_hooks.TimerHook().name == 'TimerHook'