def test_time_range(self):
push_patch = mock.patch('cupy.cuda.nvtx.RangePush')
pop_patch = mock.patch('cupy.cuda.nvtx.RangePop')
with push_patch as push, pop_patch as pop:
with profiler.time_range('test:time_range', color_id=-1):
pass
push.assert_called_once_with('test:time_range', -1)
pop.assert_called_once_with()
def test_time_range_with_ARGB(self):
push_patch = mock.patch('cupy.cuda.nvtx.RangePushC')
pop_patch = mock.patch('cupy.cuda.nvtx.RangePop')
with push_patch as push, pop_patch as pop:
with profiler.time_range('test:time_range_with_ARGB',
argb_color=0xFF00FF00):
pass
push.assert_called_once_with('test:time_range_with_ARGB',
0xFF00FF00)
pop.assert_called_once_with()
def test_time_range_err(self):
push_patch = mock.patch('cupy.cuda.nvtx.RangePush')
pop_patch = mock.patch('cupy.cuda.nvtx.RangePop')
with push_patch as push, pop_patch as pop:
try:
with profiler.time_range('test:time_range_error', -1):
raise Exception()
except Exception:
pass
push.assert_called_once_with('test:time_range_error', -1)
pop.assert_called_once_with()
def test_time_range_decorator(self):
with self.assertRaises(RuntimeError):
profiler.time_range()
def test_time_range(self):
with self.assertRaises(RuntimeError):
with profiler.time_range(''):
pass