Searched defs:NCCLTraceBuffer (Results 1 – 1 of 1) sorted by relevance
567 struct NCCLTraceBuffer { struct568 static NCCLTraceBuffer* get() { in get()574 NCCLTraceBuffer() { in NCCLTraceBuffer() function579 using Event = at::cuda::CUDAEvent;580 struct Entry {638 bool enabled_ = false;639 bool capture_cpp_stack_ = false;640 std::mutex mutex_;641 std::vector<Entry> entries_;642 size_t max_entries_ = 0;[all …]