add py_module argument to GCC_compiler.compile_str

theano/gof/cmodule.py

@@ -1483,7 +1483,7 @@ class GCC_compiler(object):
     @staticmethod
     def compile_str(module_name, src_code, location=None,
                     include_dirs=None, lib_dirs=None, libs=None,
-                    preargs=None):
+                    preargs=None, py_module=True):
         """
         :param module_name: string (this has been embedded in the src_code
 
@@ -1503,7 +1503,11 @@ class GCC_compiler(object):
 
         :param preargs: a list of extra compiler arguments
 
+        :param py_module: if False, compile to a shared library, but do not
+            import it as a Python module.
+
         :returns: dynamically-imported python module of the compiled code.
+            (unless py_module is False, in that case returns None.)
         """
         #TODO: Do not do the dlimport in this function
 
@@ -1628,6 +1632,7 @@ class GCC_compiler(object):
             # Print errors just below the command line.
             print compile_stderr
 
+        if py_module:
             #touch the __init__ file
             file(os.path.join(location, "__init__.py"), 'w').close()
             return dlimport(lib_filename)