Let's Write an LLVM Specializer for Python(dev.stephendiehl.com)

287 pointsby tenslisi12 anni fa43 comments

dev.stephendiehl.com

Let's Write an LLVM Specializer for Python

http://dev.stephendiehl.com/numpile/

44 comments

This is a great tutorial about first-generation Numba. The author learned a lot about LLVM and llvmpy while working with several of our devs. If you are interested in the "Further work" in his article, come join the Numba project.

tadlan12 anni fa

What is second generation numba? And any plans to branch numba out of pure numeric application s?

wedesoft12 anni fa

I did something similar in Ruby but using GCC as a "JIT" compiler for image processing (software [1], thesis [2]). I can really recommend JIT compilation for doing array processing.

[1] http://www.wedesoft.de/hornetseye-api/ [2] http://www.wedesoft.de/downloads/thesis_wedekind.pdf

EDIT: In my approach I didn't go through the Ruby AST though. Rather I used the approach of injecting "GCCVariables" which emit C code instead of doing the actual computation.

chrisseaton12 anni fa

You should submit your thesis to the Ruby Bibliography http://rubybib.org

wedesoft12 anni fa

Ok, will do. Cheers :)

ericfrederich12 anni fa

I love Python but when I use a statically typed language like C, Rust, Go, etc I really feel that it is missing from Python.

I'd love to see a new language exactly like Python but compiled and statically typed. Something similar to Cython, but rather than generating a bunch of C code it would target LLVM. Additionally it would be able to generate pure Python code simply by removing any typing syntax.

cyberneticcook12 anni fa

Could this be used to ahead-of-time compile Python code ? I'm more interested in getting to a native executable or library.

walkamages12 anni fa

An excellent article! I had wanted to get back into some python recently after seeing the changes in 3.4, I had also wanted to become more familiar with LLVM, and this does both.

gtirloni12 anni fa

there is this discussion (flame war?) about python3 not bringing too many benefits. i haven't made my mind yet. could you elaborate what you saw in 3.4 that was nice?

ak21712 anni fa

Without listing any of the modules or improvements that are in the standard library in 3 but backported as PyPI modules to 2 (of which there are many), here are the features that I actually use in Python 3: unicode handling that isn't insane, function annotations, async improvements, exception chaining, enums, single-dispatch generics, better SSL support, generator delegation, better int-bytes conversion support, unittest module improvements.

The key point is that 2.7 is a language frozen in time, while 3.4+ is continuing to develop and improve. And most of the hand-wringing was before the critical mass of third-party modules was ported to 3.x.

https://docs.python.org/3/whatsnew/3.4.html

https://docs.python.org/3/whatsnew/3.3.html

https://docs.python.org/3/whatsnew/3.2.html

https://docs.python.org/3/whatsnew/3.1.html

https://docs.python.org/3/whatsnew/3.0.html

ngoldbaum12 anni fa

My favorite new feature is PEP-442 [0]. Basically, it's now safe to add a __del__ method to a class without worrying about memory leaks caused by reference cycles.

[0] https://www.python.org/dev/peps/pep-0442/

pekk12 anni fa

Most people should not be writing __del__ methods at all, especially if what they are trying to do is deterministic cleanup.

andreasvc12 anni fa

I can't say I disagree with such sentiments, but over time I've ran into a few issues (features and performance improvements) which where only addressed in Python 3. This is reason enough to try working with Python 3.

bkeroack12 anni fa

Add to the list:

- function annotations (allows runtime type checking via third party modules)

- asyncio (not as easy to use as Go's goroutines but still vastly superior to the multiprocessing module)

gamesbrainiac12 anni fa

I'd like to know too. Are there any LLVM specific enhancements that python 3.4 brings to the table?

exDM6912 anni fa

No. AFAIK, there's nothing related to LLVM in core Python. And not in 3.4 changes either.

exDM6912 anni fa

Simply put: it's a better language. The whole "discussion" is whether it makes sense to migrate since many parts of the ecosystem (many important libraries and frameworks) have not made the transition. And many distros ship Python 2 by default, Python 3 is optional. Python 3 only is not feasible.

To me, the killer feature is better lazy evaluation (generators). In particular, important builtins like map, filter, zip, enumerate, etc are generators, instead of returning lists. This makes it feasible to write things like

    (process(line) for line in map(str.upper, open('giantfile.txt')) if line.lstrip()[0] != '#')

Some of the above can also be done with itertools package in Python 2, but not everything.

Python 3.4 changelog is here, it contains e.g. asynchronous io facilities (asyncio module): https://www.python.org/downloads/release/python-342/

edit: added enumerate() in the example above, for line in open(filename) returns a generator in Python 2.x too.

edit2: enumerate is lazy in python2, I replaced it with map(str.upper)

halflings12 anni fa

The example you gave works perfectly in Python 2.7 (would also be a generator, and you're not using map filter or else); but I agree: those should've been generators from day 1, especially zip and enumerate since they make more elegant code but often come with a performance overhead in Python 2.7

maxerickson12 anni fa

Python 2 enumerate returns an 'enumerate' object that is more or less a light weight wrapper of the sequence that was passed in.

Generators provide a convenient syntax to implement that sort of object.

exDM6912 anni fa

D'oh. I put in a map(), that returns a list.

danohuiginn12 anni fa

Turning them into generators would have broken a lot of existing code, though, so it's reasonable to leave them until a major version change. Rather, these are the kind of small, obviously-useful changes that should have come immediately in 3.0, giving people some encouragement to switch.

exDM6912 anni fa

Most of the breaking changes (including making map, zip, filter, etc lazy) were done in Python 3.0.

exDM6912 anni fa

Ah, you're absolutely right. for line in open(filename) returns a generator. I added map(str.upper, xxx) there to make it more it make sense.

But the point should be obvious, without generators that would potentially consume a lot of memory.

danohuiginn12 anni fa

        (process(line) for line in open('giantfile.txt') if line.lstrip()[0] != '#')

Is that line really using any new features in Python 3? The lazy evaluation there is in the file object and the generator expression, both of which have long been present in python2.

exDM6912 anni fa

No, it wasn't. I added map(str.upper, xxx), now it does.

masklinn12 anni fa

Then again a lazy map is just an import away in P2. It removes pitfalls from Python but the improvement is… very limited (as opposed to e.g. `yield from` which is a big convenience, or for much more specialised uses the ellipsis literal being universal)

dalke12 anni fa

  (process(line.upper()) for line in open('giantfile.txt') if line.lstrip()[0] != '#')

exDM6912 anni fa

That's besides the point. It's trivial to write that as a for loop or in a million different ways to avoid the issue. It's a contrived example written to demonstrate a difference.

Here's another one you can't change that easily:

    tests_pass = all(process(input) == output for (input, output) in zip(open('inputs.txt'), open('outputs.txt'))

andreasvc12 anni fa

You can change that easily, with izip from itertools.

The fact that a bunch of builtins and the values/items methods of dictionaries have become iterators is not very siginificant IMHO. Python 2 code could already be written to use iterators or generator expressions, so in the parts where it was crucial it was already done. In this regard Python 3 has not added new functionality but only changed defaults.

The unicode change is the big one.

dalke12 anni fa

In this case (checking that process(input) == output), itertools.izip_longest is probably that right solution, unless there's an out-of-band way to know that inputs.txt is the same length as outputs.txt.

dalke12 anni fa

You felt the need to correct yourself earlier, so I think your "besides the point" should be directed to yourself. I was pointing out that your correction wasn't persuasive.

smazga12 anni fa

We went to Python 3 for the multiprocessing module. At the time it was 3.3, but now 3.4 has all that async magic. I wish I still worked on that project.

chrisheller12 anni fa

You'll get an IndexError exception on that if there are any blank lines in the file.

Changing that to line.lstrip().startswith('#') would be an alternate approach.

exDM6912 anni fa

You're right but that is irrelevant, it's a somewhat contrived example anyway. It's not like I spent a lot of time trying it out.

mkesper12 anni fa

Saner handling of Unicode, for example.

chc12 anni fa

This alone is pretty wonderful. I've only been working in Python a few months and the number of issues I've had to debug in 2.7 that came down to Unicode handling is kind of nuts.

baq12 anni fa

much better exceptions and sane unicode are the biggest improvements.

illumen12 anni fa

Very nice article! :)

Storing types via traces could be another step for gathering types. As well as using the more advanced static type checking code that is around for python.

Now I have something to work through on the weekend. Looking forward to part 2!

travisoliphant12 anni fa

The numba code-base implements quite a bit of this. We actually moved away from the AST approach and went back to the byte-code approach because the AST approach quickly becomes unwieldy as the number of Visitors that you apply grows. Compile times are also slower.

The author is definitely helping people learn about LLVM and how it can be used with Python --- which is great, because this is exactly what Numba is: http://numba.pydata.org. But, please don't start another "Numba". Just come help us improve the current one.

Erwin12 anni fa

Interesting. I currently use Python's AST to convert some nested logical query expression (in a syntax unique to my application) into bytecode executed by a specialized VM (I originally tried using V8 and LuaJit for this but performance wise that was unsuccessful; the project replaced some old Boost::Python C++ code). This article should make it easy to get started attempting an LLVM replacement.

travisoliphant12 anni fa

Yes, LLVM is a great approach for doing code-gen from arbitrary specialized VMs. And the Python interface to it makes it easy to experiment. We no longer use llvmpy for Numba (we use a simpler interface llvmlite) and so llvmpy could use a maintainer.

[deleted]12 anni fa

jonstewart12 anni fa

I really appreciate the length and detail in this blog post. It's comprehensive, not just showing off.

ch0wn12 anni fa

Stephen Diehl continues to blow my mind on a regular basis. His latest work in progress "Wrote You a Haskell"[0] is also worth keeping an eye on. I've worked through the first couple of chapters and it's fantastic.

[0] http://dev.stephendiehl.com/fun/