From 0ffaf04964034ee6a01a5ac15ceac35746a2ee9e Mon Sep 17 00:00:00 2001
From: Wangzheee <634486483@qq.com>
Date: Wed, 8 Jun 2022 13:47:22 +0800
Subject: [PATCH] fix_ernie_unitest (#43283)

---
 paddle/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddle/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc b/paddle/fluid/inference/tests/infer_ut/test_ernie_xnli_int8.cc
index 53edc554eb..779b6ee8ee 100644
--- 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);
 }
-- 
GitLab