commit d51766100ced776996224be057281f9763549c28
author Boian Petkantchin <boian.petkantchin@amd.com> Tue Jan 07 07:46:45 2025 -0800
committer GitHub <noreply@github.com> Tue Jan 07 15:46:45 2025 +0000
tree 45b2304c1bbbbab3159d3cf67a6fda27cf1198a2
parent d2242207764230ad398585a5771f9d54ce91b4c8

[runtime][python] Add debug sink to bindings (#19013)

We don't support custom debug sinks in the Runtime Python bindings.
In particular the ability to register a custom callback when tracing
tensors.

This change makes it possible to create a HAL module with a Python
function as a callback.
This implementation does not handle the case of referencing directly or
indirectly the HAL module, VM context or VM instance in the callback
function object. In such a scenario the circular reference will not be
collected by the garbage collector and will leak. No no check is done
to guard against this. It is possible to traverse the Python object
structure to detect a reference to VM objects but it would require more
effort.

Here is added a callback to the debug sink in the IREE native runtime
API that signals when the runtime is done using the debug sink.
We need this since the Python objects corresponding to native runtime
objects are ephemeral and can not be used to hold the reference to the
debug sink.

---------

Signed-off-by: Boian Petkantchin <boian.petkantchin@amd.com>