Skip to content

Commit

Permalink
Fix #4520
Browse files Browse the repository at this point in the history
  • Loading branch information
sklam committed Sep 26, 2019
1 parent 85d5cf5 commit 5a79689
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions numba/typedobjectutils.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,16 +128,23 @@ def _get_incref_decref(context, module, datamodel, container_type):
refct_fnty,
name='.numba_{}_incref${}'.format(container_type, fe_type),
)

builder = ir.IRBuilder(incref_fn.append_basic_block())
context.nrt.incref(builder, fe_type, builder.load(incref_fn.args[0]))
context.nrt.incref(
builder, fe_type,
datamodel.load_from_data_pointer(builder, incref_fn.args[0]),
)
builder.ret_void()

decref_fn = module.get_or_insert_function(
refct_fnty,
name='.numba_{}_decref${}'.format(container_type, fe_type),
)
builder = ir.IRBuilder(decref_fn.append_basic_block())
context.nrt.decref(builder, fe_type, builder.load(decref_fn.args[0]))
context.nrt.decref(
builder, fe_type,
datamodel.load_from_data_pointer(builder, decref_fn.args[0]),
)
builder.ret_void()

return incref_fn, decref_fn
Expand Down

0 comments on commit 5a79689

Please sign in to comment.