From a5a8558feb9417359e30a991ab5e01cf17194473 Mon Sep 17 00:00:00 2001
From: Alexandre Passos <apassos@google.com>
Date: Wed, 12 Apr 2017 15:30:57 -0800
Subject: Fetch / py_func arguments no longer copy. Change: 153003331

---
 tensorflow/c/c_api_test.cc | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'tensorflow/c/c_api_test.cc')

diff --git a/tensorflow/c/c_api_test.cc b/tensorflow/c/c_api_test.cc
index 5673f657d3..32143f4f2f 100644
--- a/tensorflow/c/c_api_test.cc
+++ b/tensorflow/c/c_api_test.cc
@@ -105,6 +105,22 @@ TEST(CAPI, AllocateTensor) {
   TF_DeleteTensor(t);
 }
 
+TEST(CAPI, MaybeMove) {
+  const int num_bytes = 6 * sizeof(float);
+  float* values =
+      reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
+          EIGEN_MAX_ALIGN_BYTES, num_bytes));
+  int64_t dims[] = {2, 3};
+  bool deallocator_called = false;
+  TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes,
+                              &Deallocator, &deallocator_called);
+
+  TF_Tensor* o = TF_TensorMaybeMove(t);
+  ASSERT_TRUE(o == nullptr);  // It is unsafe to move memory TF might not own.
+  TF_DeleteTensor(t);
+  EXPECT_TRUE(deallocator_called);
+}
+
 TEST(CAPI, LibraryLoadFunctions) {
   // Load the library.
   TF_Status* status = TF_NewStatus();
-- 
cgit v1.2.3