mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-02 23:39:58 +00:00
27 lines
482 B
C++
27 lines
482 B
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
|
|
namespace onnxruntime {
|
|
namespace perftest {
|
|
namespace utils {
|
|
|
|
size_t GetPeakWorkingSetSize();
|
|
|
|
class ICPUUsage {
|
|
public:
|
|
virtual ~ICPUUsage() = default;
|
|
|
|
virtual short GetUsage() const = 0;
|
|
|
|
virtual void Reset() = 0;
|
|
};
|
|
|
|
std::unique_ptr<ICPUUsage> CreateICPUUsage();
|
|
|
|
} // namespace utils
|
|
} // namespace perftest
|
|
} // namespace onnxruntime
|