[tensorflow-sys] add status check code.

tensorflow · Nov 11, 2021 · 34815b5 · 34815b5
1 parent aa9f6cc
commit 34815b5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tensorflow-sys/tests/lib.rs b/tensorflow-sys/tests/lib.rs
@@ -32,6 +32,14 @@ fn tfe_tensor_handle() {
         );
         let status = ffi::TF_NewStatus();
         let tfe_handle = ffi::TFE_NewTensorHandle(tf_tensor, status);
+        if ffi::TF_GetCode(status) != ffi::TF_OK {
+            panic!(
+                "{}",
+                std::ffi::CStr::from_ptr(ffi::TF_Message(status))
+                    .to_string_lossy()
+                    .into_owned()
+            );
+        }
 
         ffi::TF_DeleteStatus(status);
         ffi::TFE_DeleteTensorHandle(tfe_handle);