Skip to content
Snippets Groups Projects

Reveal base pointer spec

Merged Markus Holzer requested to merge holzer/pystencils:BasePointerSpec into master
Files
7
@@ -73,7 +73,11 @@ def create_kernel(assignments: NodeCollection,
@@ -73,7 +73,11 @@ def create_kernel(assignments: NodeCollection,
typed_split_groups = [[type_symbol(s) for s in split_group] for split_group in split_groups]
typed_split_groups = [[type_symbol(s) for s in split_group] for split_group in split_groups]
split_inner_loop(ast_node, typed_split_groups)
split_inner_loop(ast_node, typed_split_groups)
base_pointer_spec = [['spatialInner0'], ['spatialInner1']] if len(loop_order) >= 2 else [['spatialInner0']]
base_pointer_spec = config.base_pointer_specification
 
if base_pointer_spec is None:
 
base_pointer_spec = []
 
if config.cpu_vectorize_info and config.cpu_vectorize_info.get('nontemporal'):
 
base_pointer_spec = [['spatialInner0'], ['spatialInner1']] if len(loop_order) >= 2 else [['spatialInner0']]
base_pointer_info = {field.name: parse_base_pointer_info(base_pointer_spec, loop_order,
base_pointer_info = {field.name: parse_base_pointer_info(base_pointer_spec, loop_order,
field.spatial_dimensions, field.index_dimensions)
field.spatial_dimensions, field.index_dimensions)
for field in fields_without_buffers}
for field in fields_without_buffers}
Loading