未验证 提交 19438131 编写于 作者: W Wangzheee 提交者: GitHub

[Paddle Inference]test=infer-coverage (#46955)

* test=infer-coverage
上级 20335b7c
...@@ -23,6 +23,9 @@ namespace paddle { ...@@ -23,6 +23,9 @@ namespace paddle {
namespace inference { namespace inference {
void run(const AnalysisConfig& config, std::vector<float>* out_data, int bs) { void run(const AnalysisConfig& config, std::vector<float>* out_data, int bs) {
#if !defined(_WIN32)
setenv("NVIDIA_TF32_OVERRIDE", "0", 1);
#endif
auto predictor = CreatePaddlePredictor(config); auto predictor = CreatePaddlePredictor(config);
auto input_names = predictor->GetInputNames(); auto input_names = predictor->GetInputNames();
...@@ -222,6 +225,9 @@ std::shared_ptr<paddle_infer::Predictor> InitPredictor() { ...@@ -222,6 +225,9 @@ std::shared_ptr<paddle_infer::Predictor> InitPredictor() {
} }
void run(paddle_infer::Predictor* predictor, std::vector<float>* out_data) { void run(paddle_infer::Predictor* predictor, std::vector<float>* out_data) {
#if !defined(_WIN32)
setenv("NVIDIA_TF32_OVERRIDE", "0", 1);
#endif
const int run_batch = 2; const int run_batch = 2;
const int run_seq_len = 71; const int run_seq_len = 71;
const int max_seq_len = 128; const int max_seq_len = 128;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册