Searched refs:CuptiTracerEventType (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
D | cupti_tracer.cc | 141 CuptiTracerEventType MemcpyKind(const CudaMemcpy *p) { in MemcpyKind() 144 return CuptiTracerEventType::MemcpyH2D; in MemcpyKind() 148 return CuptiTracerEventType::MemcpyD2H; in MemcpyKind() 152 return CuptiTracerEventType::MemcpyD2D; in MemcpyKind() 154 return CuptiTracerEventType::Unsupported; in MemcpyKind() 157 std::tuple<size_t /*bytes*/, CuptiTracerEventType, bool /*async*/> 162 return {p->ByteCount, CuptiTracerEventType::MemcpyH2D, false}; in DecodeDriverMemcpy() 167 return {p->ByteCount, CuptiTracerEventType::MemcpyH2D, true}; in DecodeDriverMemcpy() 171 return {p->ByteCount, CuptiTracerEventType::MemcpyD2H, false}; in DecodeDriverMemcpy() 176 return {p->ByteCount, CuptiTracerEventType::MemcpyD2H, true}; in DecodeDriverMemcpy() [all …]
|
D | cupti_collector.cc | 46 if (event.type != CuptiTracerEventType::Overhead) { in IsHostEvent() 177 if (event.type == CuptiTracerEventType::Kernel && in CreateXEvent() 211 } else if (event.type == CuptiTracerEventType::MemcpyH2D || in CreateXEvent() 212 event.type == CuptiTracerEventType::MemcpyD2H || in CreateXEvent() 213 event.type == CuptiTracerEventType::MemcpyD2D || in CreateXEvent() 214 event.type == CuptiTracerEventType::MemcpyP2P || in CreateXEvent() 215 event.type == CuptiTracerEventType::MemcpyOther) { in CreateXEvent() 226 } else if (event.type == CuptiTracerEventType::MemoryAlloc) { in CreateXEvent() 234 } else if (event.type == CuptiTracerEventType::MemoryFree) { in CreateXEvent() 242 } else if (event.type == CuptiTracerEventType::Memset) { in CreateXEvent() [all …]
|
D | cupti_collector.h | 113 enum class CuptiTracerEventType { enum 130 const char* GetTraceEventTypeName(const CuptiTracerEventType& type); 148 CuptiTracerEventType type = CuptiTracerEventType::Unsupported;
|