Hi,
Yes a simple cuda extension with the new cpp extension should be very simple to do. And the easiest way to do this.
@smth There is definitly no api that lets you do that easily. I’ll take a look when I have a bit of free time and I’m done with the hook thing.