Auto-generation of basic ufuncs with numba

[sklam]

@skrah

Here're the scripts to generate a .o file with basic ufuncs supported by numba:
https://gist.github.com/sklam/977dd295324db10868542567f526ecf6

Usage:

• python ufunc_gen.py to create umath.o.
• run cc -shared -o umath.so umath.o.
• test with python check_ufunc_gen.py.

See top of file comments for details like ABI and symbol mangling.

[skrah]

@sklam This is fantastic, thanks!

[skrah]

The non-strided loops work nicely here. As I understood, strided loops in the last dimension are not yet supported in Numba. How difficult would it be to add that?

For example, these tests would also pass:

https://github.com/plures/gumath/blob/master/python/test_gumath.py#L89

[sklam]

I thought the numba generated kernel doesn't care about the striding. It should always read the strides from the argument. Let me check.

[sklam]

You're right. It should be easy to fix.

[sklam]

I have updated the gist with a new script. There was a mistake in using the wrong stride.