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

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

* test=infer-coverage
上级 20335b7c
......@@ -23,6 +23,9 @@ namespace paddle {
namespace inference {
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 input_names = predictor->GetInputNames();
......@@ -222,6 +225,9 @@ std::shared_ptr<paddle_infer::Predictor> InitPredictor() {
}
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_seq_len = 71;
const int max_seq_len = 128;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册