Skip to content
Snippets Groups Projects

Fix nontemporal stores on non-x86 for fields with variable size

Merged Michael Kuron requested to merge nontemporal into master
Files
2
+ 7
0
@@ -619,7 +619,14 @@ class SympyAssignment(Node):
@@ -619,7 +619,14 @@ class SympyAssignment(Node):
for i in range(len(symbol.offsets)):
for i in range(len(symbol.offsets)):
loop_counters.add(LoopOverCoordinate.get_loop_counter_symbol(i))
loop_counters.add(LoopOverCoordinate.get_loop_counter_symbol(i))
result.update(loop_counters)
result.update(loop_counters)
 
result.update(self._lhs_symbol.atoms(sp.Symbol))
result.update(self._lhs_symbol.atoms(sp.Symbol))
 
 
sizes = set().union(*(a.field.shape[:a.field.spatial_dimensions]
 
for a in self._lhs_symbol.atoms(ResolvedFieldAccess)))
 
sizes = filter(lambda s: isinstance(s, FieldShapeSymbol), sizes)
 
result.update(sizes)
 
return result
return result
@property
@property
Loading