Use memref.atomic_rmw instead.

xdslproject · Jun 10, 2024 · fad6922 · fad6922
1 parent 4972b91
commit fad6922
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/.github/workflows/ci-mlir-mpi.yml b/.github/workflows/ci-mlir-mpi.yml
@@ -36,7 +36,7 @@ jobs:
       run: |
         pip install -e .[tests]
         pip install mpi4py
-        pip install git+https://github.com/xdslproject/xdsl@210181350d926f91ee5fdb27f0eb5d1cf53a8997
+        pip install git+https://github.com/xdslproject/xdsl@1ffdc2d9f33a435af7382a668e153fade01a1255
 
     - name: Test with MPI
       run: |

diff --git a/.github/workflows/ci-mlir.yml b/.github/workflows/ci-mlir.yml
@@ -36,7 +36,7 @@ jobs:
       run: |
         pip install -e .[tests]
         pip install mpi4py
-        pip install git+https://github.com/xdslproject/xdsl@210181350d926f91ee5fdb27f0eb5d1cf53a8997
+        pip install git+https://github.com/xdslproject/xdsl@1ffdc2d9f33a435af7382a668e153fade01a1255
         
     - name: Test no-MPI, no-Openmp
       run: |

diff --git a/devito/ir/ietxdsl/cluster_to_ssa.py b/devito/ir/ietxdsl/cluster_to_ssa.py
@@ -369,10 +369,9 @@ def build_generic_step_expression(self, dim: SteppingDimension, eq: LoweredEq):
 
         match eq.operation:
             case None:
-                pass
+                memref.Store.get(value, memtemp, ssa_indices)
             case OpInc:
-                value = arith.Addf(value, memref.Load.get(memtemp, ssa_indices).res)
-        memref.Store.get(value, memtemp, ssa_indices)
+                memref.AtomicRMWOp(operands=[value, memtemp, ssa_indices], result_types=[value.type], properties={"kind" : builtin.IntegerAttr(0, builtin.i64)})
 
     def build_condition(self, dim: SteppingDimension, eq: BooleanFunction):
         return self._visit_math_nodes(dim, eq, None)