fix

Author: co63oc

tests/operators/test_speculate_get_output_padding_offset.py

@@ -29,7 +29,7 @@ def test_speculate_get_output_padding_offset(self):
         output_token_num = np.sum(seq_lens_output)
 
         seq_lens_output = paddle.to_tensor(seq_lens_output, dtype="int32")
-        out_token_num = paddle.sum(seq_lens_output)
+        out_token_num = paddle.sum(seq_lens_output).astype("int32")
         output_cum_offsets_tmp = paddle.cumsum(max_seq_len - seq_lens_output)
 
         output_padding_offset_gpu, output_cum_offsets_gpu = speculate_get_output_padding_offset(

tests/operators/test_speculate_get_token_penalty_multi_scores.py

@@ -86,8 +86,8 @@ def _test_main(
             output_cum_offsets_ref,
             max_seq_len,
         )
-        logits_ref_np = logits_ref.astype("float32").numpy()
-        logits_np = logits.astype("float32").numpy()
+        logits_ref_np = logits_ref.clone().numpy().astype("float32")
+        logits_np = logits.clone().numpy().astype("float32")
         np.testing.assert_allclose(logits_ref_np, logits_np, atol=1e-5, rtol=1e-5)
 
     def test_speculate_get_token_penalty_multi_scores(self):