torch::autograd::profiler::CUDAStubs Struct Reference

Public Member Functions
virtual void	record (int *device, CUDAEventStub *event, int64_t *cpu_ns)
virtual float	elapsed (CUDAEventStub event, CUDAEventStub event2)
virtual void	nvtxMarkA (const char *name)
virtual void	nvtxRangePushA (const char *name)
virtual void	nvtxRangePop ()
virtual bool	enabled ()
virtual void	onEachDevice (std::function< void(int)> op)
virtual void	synchronize ()

Detailed Description

Definition at line 30 of file profiler.h.

---

The documentation for this struct was generated from the following files:

• torch/csrc/autograd/profiler.h
• torch/csrc/autograd/profiler.cpp