Add pass to convert atomic rmw to non-atomic ops when legal (#1782)

* wip remove atomics

wip

wip

testing

wip

wip

done?

file

remove unneeded files

actually run test

revert isl

dep dialects

missingheader

stray file

* 2d seems to work as well

Author: ivanradanov

src/enzyme_ad/jax/Passes/Passes.td

@@ -1159,4 +1159,12 @@ def ConvertAllConstantsToSplattedConstantPass : Pass<"convert-all-constants-to-s
   ];
 }
 
+def RemoveAtomicsPass : Pass<"remove-atomics"> {
+  let summary = "Remove unnecessary atomics";
+  let dependentDialects = [
+    "affine::AffineDialect",
+    "enzymexla::EnzymeXLADialect",
+  ];
+}
+
 #endif