fix_ernie_unitest (#43283)

0ffaf049 · Wangzheee · GitHub · 88216f63 · 0ffaf049
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc ...le/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc +4 -0

未找到文件。
--- a/paddle/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc
+++ b/paddle/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc
@@ -69,6 +69,10 @@ std::shared_ptr<Predictor> InitPredictor() {
                                opt_input_shape);
  // erinie varlen must be used with oss
  config.EnableVarseqlen();
+  paddle_infer::experimental::InternalUtils::SetTransformerPosid(&config,
+                                                                 input_name2);
+  paddle_infer::experimental::InternalUtils::SetTransformerMaskid(&config,
+                                                                  input_name3);

  return CreatePredictor(config);
 }