commit 7a18c91de6272c93468f9987d02a480a61a4b38c
author Dan Moldovan <mdan@google.com> Fri Jun 04 07:49:08 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Jun 04 07:54:23 2021 -0700
tree ac0b08ba3afd54ac457cdea0ec83865e76313c16
parent b6720c3bde7d04c042b5b323298a97d0877ced8c

Fix null-pointer-use error detected by asan.

PiperOrigin-RevId: 377510821
Change-Id: Id7a1ebe10e6c2f1f2aebf69b9ce395c2f7190523

tensorflow/core/framework/function_test.cc

diff --git a/tensorflow/core/framework/function_test.cc b/tensorflow/core/framework/function_test.cc
index cafe343..be6403a 100644
--- a/tensorflow/core/framework/function_test.cc
+++ b/tensorflow/core/framework/function_test.cc
@@ -914,7 +914,10 @@
   HasError(frame.SetArgs({a}), "Invalid argument");
   const Tensor* v;
   HasError(frame.GetArg(0, &v), "Invalid argument");
-  HasError(frame.SetRetval(0, *v), "Invalid argument");
+  if (v != nullptr) {
+    // v is null in certain environments.
+    HasError(frame.SetRetval(0, *v), "Invalid argument");
+  }
   std::vector<Tensor> rets;
   TF_EXPECT_OK(frame.GetRetvals(&rets));
   EXPECT_EQ(rets.size(), 0);