tensorplay.cuda.profiler.profile

tensorplay.cuda.profiler.profile()[source]

Enable profiling.

Context Manager to enabling profile collection by the active profiling tool from CUDA backend. .. rubric:: Example

>>> # xdoctest: +REQUIRES(env:TORCH_DOCTEST_CUDA)
>>> import tensorplay as tp
>>> model = tp.nn.Linear(20, 30).cuda()
>>> inputs = tp.randn(128, 20).cuda()
>>> with tp.cuda.profiler.profile():
...     model(inputs)