search

Nicholaswogan / numbalsoda

Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.
MIT License
91 stars 9 forks source link

Make NumPy types for u0 and usol configurable #17

Open juhannc opened 2 years ago

juhannc commented 2 years ago

First of all, awesome code! I got a speedup of an order of magnitude basically for free!

But, as far as I can tell, u0 and usol are required to be of type np.float64. However, often times np.float32 is more than enough.

Maybe I just don't see the option to change it, but if it doesn't exist, it'd be nice if we could change it.

Thank you!

Nicholaswogan commented 2 years ago

Thanks! Ya, it should be possible to just compile the underlying C++ code for either float64 or float32. I will leave this open and get it to eventually.