Commit c1f3fd28 authored by da-woods's avatar da-woods Committed by GitHub

Release temps used for buffer indexing after use (GH-3517)

Closes https://github.com/cython/cython/issues/3430
parent 34cde7c6
......@@ -4331,7 +4331,7 @@ class BufferIndexNode(_IndexingBaseNode):
else:
negative_indices = Buffer.buffer_defaults['negative_indices']
return buffer_entry, Buffer.put_buffer_lookup_code(
buffer_lookup_result = Buffer.put_buffer_lookup_code(
entry=buffer_entry,
index_signeds=[ivar.type.signed for ivar in self.indices],
index_cnames=index_temps,
......@@ -4340,6 +4340,9 @@ class BufferIndexNode(_IndexingBaseNode):
negative_indices=negative_indices,
in_nogil_context=self.in_nogil_context)
# must return index_temps since that cannot be released until buffer_lookup_result has been used
return buffer_entry, buffer_lookup_result, index_temps
def generate_assignment_code(self, rhs, code, overloaded_assignment=False):
self.generate_subexpr_evaluation_code(code)
self.generate_buffer_setitem_code(rhs, code)
......@@ -4370,7 +4373,7 @@ class BufferIndexNode(_IndexingBaseNode):
return
# Used from generate_assignment_code and InPlaceAssignmentNode
buffer_entry, ptrexpr = self.buffer_lookup_code(code)
buffer_entry, ptrexpr, buffer_temps = self.buffer_lookup_code(code)
if self.buffer_type.dtype.is_pyobject:
# Must manage refcounts. Decref what is already there
......@@ -4388,6 +4391,8 @@ class BufferIndexNode(_IndexingBaseNode):
else:
# Simple case
code.putln("*%s %s= %s;" % (ptrexpr, op, rhs.result()))
for temp in buffer_temps:
code.funcstate.release_temp(temp)
def generate_result_code(self, code):
if is_pythran_expr(self.base.type):
......@@ -4399,13 +4404,15 @@ class BufferIndexNode(_IndexingBaseNode):
self.base.pythran_result(),
pythran_indexing_code(self.indices)))
return
buffer_entry, self.buffer_ptr_code = self.buffer_lookup_code(code)
buffer_entry, self.buffer_ptr_code, buffer_temps = self.buffer_lookup_code(code)
if self.type.is_pyobject:
# is_temp is True, so must pull out value and incref it.
# NOTE: object temporary results for nodes are declared
# as PyObject *, so we need a cast
code.putln("%s = (PyObject *) *%s;" % (self.result(), self.buffer_ptr_code))
code.putln("__Pyx_INCREF((PyObject*)%s);" % self.result())
for temp in buffer_temps:
code.funcstate.release_temp(temp)
class MemoryViewIndexNode(BufferIndexNode):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment