saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-10 17:37:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Test disabling profiler

Browse source
This commit is contained in:
Ryan Hill 2021-04-24 02:43:06 -07:00
parent 393286f484
commit 8c1b52cf6e
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/common/profiler.cc
View file

@ -27,13 +27,14 @@ class DeviceProfiler {
(((uintptr_t)(buffer) & ((align)-1)) ? ((buffer) + (align) - ((uintptr_t)(buffer) & ((align)-1))) : (buffer))
#define DUR(s, e) std::lround(static_cast<double>(e - s) / 1000)
class CudaProfiler final: public DeviceProfiler {
class CudaProfiler final : public DeviceProfiler {
public:
friend class DeviceProfiler;
~CudaProfiler() = default;
ORT_DISALLOW_COPY_ASSIGNMENT_AND_MOVE(CudaProfiler);
void StartProfiling(TimePoint start_time, int pid, int tid) override;
std::vector<EventRecord> EndProfiling() override;
private:
CudaProfiler() = default;
static void CUPTIAPI BufferRequested(uint8_t**, size_t*, size_t*);
@ -133,10 +134,11 @@ std::vector<EventRecord> CudaProfiler::EndProfiling() {
enabled_.clear();
return events;
}
#endif //USE_CUDA
#endif //USE_CUDA
DeviceProfiler* DeviceProfiler::GetDeviceProfiler() {
#ifdef USE_CUDA
#if 0
//#ifdef USE_CUDA
static CudaProfiler cuda_profiler;
return &cuda_profiler;
#else