o ei@sJddlZddlmZmZgdZgdZddZdd Zejd d Z dS) N) check_errorcudart)startstopprofile)gpustarttimestampgpuendtimestamp gridsize3dthreadblocksizestreamidzenableonstart 0conckerneltracecCttdS)zStarts cuda profiler data collection. .. warning:: Raises CudaError in case of it is unable to start the profiler. N)rrcudaProfilerStartrr]/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/cuda/profiler.pyrrcCr)zStops cuda profiler data collection. .. warning:: Raises CudaError in case of it is unable to stop the profiler. N)rrcudaProfilerStoprrrrrrrccs$z tdVWtdStw)a Enable profiling. Context Manager to enabling profile collection by the active profiling tool from CUDA backend. Example: >>> # xdoctest: +REQUIRES(env:TORCH_DOCTEST_CUDA) >>> import torch >>> model = torch.nn.Linear(20, 30).cuda() >>> inputs = torch.randn(128, 20).cuda() >>> with torch.cuda.profiler.profile() as prof: ... model(inputs) N)rrrrrrr&s r) contextlibrr__all__ DEFAULT_FLAGSrrcontextmanagerrrrrrs