Functions 3
profile
functionFull reference ↗- tensorplay.cuda.profiler.profile()[source]
Enable profiling.
Context Manager to enabling profile collection by the active profiling tool from CUDA backend. .. rubric:: Example
>>> import tensorplay as tp >>> model = tp.nn.Linear(20, 30).cuda() >>> inputs = tp.randn(128, 20).cuda() >>> with tp.cuda.profiler.profile(): ... model(inputs)
start
functionFull reference ↗- tensorplay.cuda.profiler.start()[source]
Starts cuda profiler data collection.
Warning
Raises CudaError in case of it is unable to start the profiler.
stop
functionFull reference ↗- tensorplay.cuda.profiler.stop()[source]
Stops cuda profiler data collection.
Warning
Raises CudaError in case of it is unable to stop the profiler.

