demo notebook & fixes in gitlab-ci.yml

backends/cbackend.py

@@ -3,7 +3,7 @@ from sympy.utilities.codegen import CCodePrinter
 from pystencils.ast import Node
 
 
-def printCCode(astNode):
+def generateC(astNode):
     """
     Prints the abstract syntax tree as C function
     """
@@ -11,7 +11,7 @@ def printCCode(astNode):
     return printer(astNode)
 
 
-def printCudaCode(astNode):
+def generateCUDA(astNode):
     printer = CBackend(cuda=True)
     return printer(astNode)
 
@@ -60,7 +60,7 @@ class CBackend:
         self.sympyPrinter = CustomSympyPrinter()
 
     def __call__(self, node):
-        return self._print(node)
+        return str(self._print(node))
 
     def _print(self, node):
         for cls in type(node).__mro__:

cpu/__init__.py

 from pystencils.cpu.kernelcreation import createKernel, addOpenMP
 from pystencils.cpu.cpujit import makePythonFunction
-from pystencils.backends.cbackend import printCCode
+from pystencils.backends.cbackend import generateC

cpu/cpujit.py

@@ -2,7 +2,7 @@ import os
 import subprocess
 from ctypes import cdll, c_double, c_float, sizeof
 from tempfile import TemporaryDirectory
-from pystencils.backends.cbackend import printCCode
+from pystencils.backends.cbackend import generateC
 import numpy as np
 
 
@@ -67,7 +67,7 @@ def compileAndLoad(kernelFunctionNode):
             print('#include <iostream>', file=sourceFile)
             print("#include <cmath>", file=sourceFile)
             print('extern "C" { ', file=sourceFile)
-            print(printCCode(kernelFunctionNode), file=sourceFile)
+            print(generateC(kernelFunctionNode), file=sourceFile)
             print('}', file=sourceFile)
 
         compilerCmd = [CONFIG['compiler']] + CONFIG['flags'].split()