mirror of
https://github.com/Tencent/ncnn.git
synced 2026-10-28 11:47:38 +08:00
32 lines
1.2 KiB
C++
32 lines
1.2 KiB
C++
// Copyright 2026 Tencent
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
#ifndef PERFUTIL_H
|
|
#define PERFUTIL_H
|
|
|
|
#include "layer.h"
|
|
#include "mat.h"
|
|
#include "option.h"
|
|
#include "paramdict.h"
|
|
|
|
ncnn::Mat PerfMat(int w, float v = 0.01f);
|
|
ncnn::Mat PerfMat(int w, int h, float v = 0.01f);
|
|
ncnn::Mat PerfMat(int w, int h, int c, float v = 0.01f);
|
|
ncnn::Mat PerfMat(int w, int h, int d, int c, float v = 0.01f);
|
|
|
|
// high-level perf entry point: benchmark a layer across all precision and GPU variations
|
|
// layer_type: ncnn layer type name (e.g. "Convolution")
|
|
// pd, weights, input(s): layer configuration
|
|
// top_blob_count: number of output blobs (for multi-input overload)
|
|
// param_fmt: printf-style format for layer-specific params (e.g. "k=%d s=%d"), can be NULL
|
|
void perf_layer(const char* layer_type, const ncnn::ParamDict& pd,
|
|
const std::vector<ncnn::Mat>& weights,
|
|
const ncnn::Mat& input, const char* param_fmt, ...);
|
|
|
|
void perf_layer(const char* layer_type, const ncnn::ParamDict& pd,
|
|
const std::vector<ncnn::Mat>& weights,
|
|
const std::vector<ncnn::Mat>& inputs, int top_blob_count,
|
|
const char* param_fmt, ...);
|
|
|
|
#endif // PERFUTIL_H
|