Skip to content
Snippets Groups Projects

Add type conversion for SP types

Closed Markus Holzer requested to merge holzer/pystencils:FixDatatypes into master
1 unresolved thread
Compare and
2 files
+ 13
1
Preferences
Compare changes
Files
2
@@ -12,7 +12,8 @@ from pystencils.gpucuda.indexing import indexing_creator_from_params
from pystencils.simp.assignment_collection import AssignmentCollection
from pystencils.stencil import direction_string_to_offset, inverse_direction_string
from pystencils.transformations import (
loop_blocking, move_constants_before_loop, remove_conditionals_in_staggered_kernel)
loop_blocking, move_constants_before_loop, remove_conditionals_in_staggered_kernel,
replace_data_type_of_typed_symbols)
def create_kernel(assignments,
@@ -88,6 +89,8 @@ def create_kernel(assignments,
split_groups = assignments.simplification_hints['split_groups']
assignments = assignments.all_assignments
assignments = replace_data_type_of_typed_symbols(assignments, data_type)
# ---- Creating ast
if target == 'cpu':
from pystencils.cpu import create_kernel