diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc index 42bf9e07998dc0..6d335ad4f81bde 100644 --- a/paddle/phi/infermeta/unary.cc +++ b/paddle/phi/infermeta/unary.cc @@ -2202,7 +2202,7 @@ void KthvalueInferMeta(const MetaTensor& x, out->set_dtype(x.dtype()); indices->set_dims(dims); indices->share_lod(x); - indices->set_dtype(x.dtype()); + indices->set_dtype(DataType::INT64); } void LogicalNotInferMeta(const MetaTensor& x, MetaTensor* out) {