Fix AttributeError on LocalBuffer.__delete__ call in python xla client

tensorflow · Mar 5, 2019 · bed882f · bed882f
1 parent b778c03
commit bed882f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow/compiler/xla/python/xla_client.py b/tensorflow/compiler/xla/python/xla_client.py
@@ -431,7 +431,9 @@ def device(self):
 
   def delete(self):
     if self.c_buffer is not None:
-      self._backend.delete_buffer(self.c_buffer)
+      # Python may have freed c_api first.
+      if c_api:
+        self._backend.delete_buffer(self.c_buffer)
       self.c_buffer = None
 
   def destructure(self):