Make sure the compilelock gets released

theano/gof/cutils.py

@@ -11,18 +11,18 @@ if os.path.exists(os.path.join(config.compiledir,'cutils_ext.so')):
 # directory. This is important to prevent multiple processes from trying to
 # compile the cutils_ext module simultaneously.
 get_lock()
-
 try:
+    try:
         from cutils_ext.cutils_ext import *
-except ImportError:
+    except ImportError:
         import cmodule
 
         code = """
-#include <Python.h>
-extern "C"{
-static PyObject *
-run_cthunk(PyObject *self, PyObject *args)
-{
+    #include <Python.h>
+    extern "C"{
+    static PyObject *
+    run_cthunk(PyObject *self, PyObject *args)
+    {
       PyObject *py_cthunk = NULL;
       if(!PyArg_ParseTuple(args,"O",&py_cthunk))
         return NULL;
@@ -38,21 +38,21 @@ run_cthunk(PyObject *self, PyObject *args)
       int failure = fn(it);
 
       return Py_BuildValue("i", failure);
-}
+    }
 
-static PyMethodDef CutilsExtMethods[] = {
+    static PyMethodDef CutilsExtMethods[] = {
         {"run_cthunk",  run_cthunk, METH_VARARGS|METH_KEYWORDS,
          "Run a theano cthunk."},
         {NULL, NULL, 0, NULL}        /* Sentinel */
-};
+    };
 
-PyMODINIT_FUNC
-initcutils_ext(void)
-{
+    PyMODINIT_FUNC
+    initcutils_ext(void)
+    {
       (void) Py_InitModule("cutils_ext", CutilsExtMethods);
-}
-}
-"""
+    }
+    }
+    """
 
         loc = os.path.join(config.compiledir, 'cutils_ext')
         if not os.path.exists(loc):
@@ -61,6 +61,7 @@ initcutils_ext(void)
         cmodule.gcc_module_compile_str('cutils_ext', code, location=loc)
         from cutils_ext.cutils_ext import *
 
-# Release lock on compilation directory.
-release_lock()
+finally:
+    # Release lock on compilation directory.
+    release_lock()