Add CudaBackend, CudaSympyPrinter
Compare changes
+ 25
− 27
@@ -32,6 +32,11 @@ __all__ = ['generate_c', 'CustomCodeNode', 'PrintNode', 'get_headers', 'CustomSy
@@ -32,6 +32,11 @@ __all__ = ['generate_c', 'CustomCodeNode', 'PrintNode', 'get_headers', 'CustomSy
@@ -52,9 +57,15 @@ def generate_c(ast_node: Node, signature_only: bool = False, dialect='c') -> str
@@ -52,9 +57,15 @@ def generate_c(ast_node: Node, signature_only: bool = False, dialect='c') -> str
@@ -141,9 +152,9 @@ class CBackend:
@@ -141,9 +152,9 @@ class CBackend:
def __init__(self, sympy_printer=None, signature_only=False, vector_instruction_set=None, dialect='c'):
@@ -164,12 +175,12 @@ class CBackend:
@@ -164,12 +175,12 @@ class CBackend:
function_arguments = ["%s %s" % (str(s.symbol.dtype), s.symbol.name) for s in node.get_parameters()]
@@ -241,10 +252,7 @@ class CBackend:
@@ -241,10 +252,7 @@ class CBackend:
@@ -292,10 +300,9 @@ class CBackend:
@@ -292,10 +300,9 @@ class CBackend:
@@ -347,22 +354,13 @@ class CustomSympyPrinter(CCodePrinter):
@@ -347,22 +354,13 @@ class CustomSympyPrinter(CCodePrinter):
return "(%s %s %s)" % (self._print(expr.args[0]), infix_functions[expr.func], self._print(expr.args[1]))
@@ -392,8 +390,8 @@ class CustomSympyPrinter(CCodePrinter):
@@ -392,8 +390,8 @@ class CustomSympyPrinter(CCodePrinter):