Skip to content
Snippets Groups Projects
Commit 43bef41c authored by Jan Hönig's avatar Jan Hönig
Browse files

Testing llvm-backend in llvmpy

parent b92821c4
Branches
Tags
No related merge requests found
...@@ -316,6 +316,15 @@ def createLatticeBoltzmannAst(updateRule=None, optimizationParams={}, **kwargs): ...@@ -316,6 +316,15 @@ def createLatticeBoltzmannAst(updateRule=None, optimizationParams={}, **kwargs):
res = createCUDAKernel(updateRule.allEquations, res = createCUDAKernel(updateRule.allEquations,
typeForSymbol='double' if optParams['doublePrecision'] else 'float', typeForSymbol='double' if optParams['doublePrecision'] else 'float',
indexingCreator=indexingCreator, ghostLayers=1) indexingCreator=indexingCreator, ghostLayers=1)
elif optParams['target'] == 'llvm':
from pystencils.llvm import createKernel
if 'splitGroups' in updateRule.simplificationHints:
splitGroups = updateRule.simplificationHints['splitGroups']
else:
splitGroups = ()
res = createKernel(updateRule.allEquations, splitGroups=splitGroups,
typeForSymbol='double' if optParams['doublePrecision'] else 'float',
ghostLayers=1)
else: else:
return ValueError("'target' has to be either 'cpu' or 'gpu'") return ValueError("'target' has to be either 'cpu' or 'gpu'")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment