fix ci failures

PaddlePaddle · Nov 9, 2023 · 8af360a · 8af360a
1 parent 99c5d30
commit 8af360a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/test/ir/inference/test_trt_convert_multihead_matmul.py b/test/ir/inference/test_trt_convert_multihead_matmul.py
@@ -985,7 +985,7 @@ def generate_trt_nodes_num():
         generate_dynamic_shape(attrs)
         self.trt_param.workspace_size = 2013265920
         self.trt_param.precision = paddle_infer.PrecisionType.Int8
-        program_config.set_input_type(np.int8)
+        program_config.set_input_type(np.float32)
         yield self.create_inference_config(), generate_trt_nodes_num(), (
             1e-3,
             1e-3,

diff --git a/test/ir/inference/test_trt_ops_fp32_mix_precision.py b/test/ir/inference/test_trt_ops_fp32_mix_precision.py
@@ -162,7 +162,6 @@ def generate_trt_nodes_num(attrs, dynamic_shape):
         # for dynamic_shape
         generate_dynamic_shape(attrs)
         self.trt_param.precision = paddle_infer.PrecisionType.Half
-        program_config.set_input_type(np.float16)
         config = self.create_inference_config()
         InternalUtils.disable_tensorrt_half_ops(
             config,