diff --git a/paddle/fluid/eager/nan_inf_utils.cc b/paddle/fluid/eager/nan_inf_utils.cc index e71ae7cf11939ad353e3544344fa1037ae786ae6..9b0005476550de05f256c85554f58f31ff649e6e 100644 --- a/paddle/fluid/eager/nan_inf_utils.cc +++ b/paddle/fluid/eager/nan_inf_utils.cc @@ -114,7 +114,9 @@ void CheckTensorHasNanOrInf(const std::string& api_name, const Tensor& tensor) { void CheckTensorHasNanOrInf(const std::string& api_name, const paddle::optional& tensor) { - CheckTensorHasNanOrInf(api_name, tensor.get()); + if (tensor) { + CheckTensorHasNanOrInf(api_name, *tensor); + } } void CheckTensorHasNanOrInf(const std::string& api_name, @@ -168,7 +170,7 @@ void CheckTensorHasNanOrInf( const std::string& api_name, const paddle::optional>& tensors) { if (tensors) { - CheckTensorHasNanOrInf(api_name, tensors.get()); + CheckTensorHasNanOrInf(api_name, *tensors); } }