Skip to content
Advertisement

Do I need to use `nogil` in Cython

I have some Cython code that I’d like to run as quickly as possible. Do I need to release the GIL in order to do this?

Let’s suppose my code is similar to this:

JavaScript

I get a whole load of error messages from the np.zeros_like line similar to:

JavaScript

Do I need to find a way of calling np.zeros_like without the GIL? Or find some other way of allocating an array that doesn’t require the GIL?


Note: this is a self-answered question designed to clear up some common misunderstanding about Cython and the GIL (although you’re welcome to answer it too, of course!).

Second note: I’ve contributed enough to Cython that I should note it here (given that I’m bringing the topic up)

Advertisement

Answer

No – you probably don’t need to release the GIL.

The basic function of the GIL (global interpreter lock) is to ensure that Python’s internal mechanisms are not subject to race conditions, by ensuring that only one Python thread is able to run at once. However, simply holding the GIL does not slow your code down.

The two (related) occasions when you should release the GIL are:

  1. Using Cython’s parallelism mechanism. The contents of a prange loop for example are required to be nogil.

  2. If you want other (external) Python threads to be able to run at the same time.

    a. if you have a large computationally/IO-intensive block that doesn’t need the GIL then it may be “polite” to release it, just to benefit users of your code who want to do multi-threading. However, this is mostly useful rather than necessary.

    b. (very, very occasionally) it’s sometimes useful to briefly release the GIL with a short with nogil: pass block. This is because Cython doesn’t release it spontaneously (unlike Python) so if you’re waiting on another Python thread to complete a task, this can avoid deadlocks. This sub-point probably doesn’t apply to you unless you’re compiling GUI code with Cython.


The sort of Cython code that can run without the GIL (no calls to Python, purely C-level numeric operations) is often the sort of code that runs efficiently. This sometimes gives people the impression that the inverse is true and the trick is releasing the GIL, rather than the actual code they’re running. Don’t be misled by this – your (single-threaded) code will run the same speed with or without the GIL.

Therefore, if you have a nice fast Numpy function that does exactly what you want quickly on a big chunk of data, but can only be called with the GIL, then just call it – no harm is done!


As a final point: even within a nogil block (for example a prange loop) you can always get the GIL back if you need it:

JavaScript

Try not to do this too often (getting/releasing it takes time, and of course only one thread can be running this block at once) but equally it’s a good way of doing small Python operations where needed.

User contributions licensed under: CC BY-SA
4 People found this is helpful
Advertisement