search

baler-collaboration / baler

Repository of Baler, a machine learning based data compression tool
https://github.com/baler-collaboration/baler.github.io
Apache License 2.0
30 stars 26 forks source link

Add the Numba, cupy support for Baler #325

Closed neogyk closed 6 months ago

neogyk commented 10 months ago

Add the jit compilation using Numba and cupy for Baler.

exook commented 8 months ago

This was just merged to the new main branch :)

exook commented 8 months ago

Running with the new jit decorators causes a lot of terminal output regarding "nopython=False" and functions being compiled in "object mode" without "forceobj=True"

exook commented 8 months ago

I changed "nopython=False" to "forceobj=True"