mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 16:48:03 +08:00
[Benchmark] Remove pmap and use mem api (#1309)
* add GPL lisence * add GPL-3.0 lisence * add GPL-3.0 lisence * add GPL-3.0 lisence * support yolov8 * add pybind for yolov8 * add yolov8 readme * add cpp benchmark * add cpu and gpu mem * public part split * add runtime mode * fixed bugs * add cpu_thread_nums * deal with comments * deal with comments * deal with comments * rm useless code * add FASTDEPLOY_DECL * add FASTDEPLOY_DECL * fixed for windows * mv rss to pss * mv rss to pss * Update utils.cc * use thread to collect mem * Add ResourceUsageMonitor * rm useless code * fixed bug * fixed typo * update ResourceUsageMonitor * fixed bug * fixed bug * add note for ResourceUsageMonitor * deal with comments * add macros * deal with comments * deal with comments * deal with comments * re-lint * rm pmap and use mem api * rm pmap and use mem api * add mem api * Add PrintBenchmarkInfo func * Add PrintBenchmarkInfo func * Add PrintBenchmarkInfo func * deal with comments --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
@@ -65,20 +65,26 @@ class FASTDEPLOY_DECL ResourceUsageMonitor {
|
||||
|
||||
private:
|
||||
void StopInternal();
|
||||
// Get current cpu memory info
|
||||
std::string GetCurrentCpuMemoryInfo();
|
||||
// Get current gpu memory info
|
||||
std::string GetCurrentGpuMemoryInfo(int device_id);
|
||||
|
||||
bool is_supported_ = false;
|
||||
bool stop_signal_ = false;
|
||||
const int sampling_interval_;
|
||||
float max_cpu_mem_ = 0.0f;
|
||||
float max_gpu_mem_ = 0.0f;
|
||||
float max_cpu_mem_ = 0.0f; // MB
|
||||
float max_gpu_mem_ = 0.0f; // MB
|
||||
float max_gpu_util_ = 0.0f;
|
||||
const int gpu_id_ = 0;
|
||||
std::unique_ptr<std::thread> check_memory_thd_ = nullptr;
|
||||
};
|
||||
|
||||
// Remove the ch characters at both ends of str
|
||||
FASTDEPLOY_DECL std::string Strip(const std::string& str, char ch = ' ');
|
||||
|
||||
// Split string
|
||||
FASTDEPLOY_DECL void Split(const std::string& s,
|
||||
std::vector<std::string>& tokens,
|
||||
char delim = ' ');
|
||||
|
||||
} // namespace benchmark
|
||||
} // namespace fastdeploy
|
||||
|
Reference in New Issue
Block a user