[Mlir-commits] [mlir] [MLIR][CAPI][Python] Add support for querying memory effect instances (PR #213459)

[llvmlistbot at llvm.org]

================
@@ -68,6 +69,26 @@ SmallVector<std::unique_ptr<Region>> unwrapRegions(intptr_t nRegions,
   return unwrappedRegions;
 }
 
+void invokeMemoryEffectInstancesCallback(
+    SmallVectorImpl<MemoryEffects::EffectInstance> &effects,
+    MlirMemoryEffectInstancesCallback callback, void *userData) {
+  SmallVector<MlirMemoryEffectInstance> wrappedEffects;
+  wrappedEffects.reserve(effects.size());
+  for (MemoryEffects::EffectInstance &effect : effects)
+    wrappedEffects.push_back(wrap(&effect));
+  callback(wrappedEffects.size(), wrappedEffects.data(), userData);
+}
+
+void appendMemoryEffectInstances(intptr_t numEffects,
+                                 MlirMemoryEffectInstance *effects,
+                                 void *userData) {
+  auto *unwrappedEffects =
+      static_cast<SmallVectorImpl<MemoryEffects::EffectInstance> *>(userData);
+  unwrappedEffects->reserve(unwrappedEffects->size() + numEffects);
+  for (intptr_t i = 0; i < numEffects; ++i)
+    unwrappedEffects->push_back(*unwrap(effects[i]));
+}
----------------
PragmaTwice wrote:

done in https://github.com/llvm/llvm-project/pull/213459/commits/b16755b3d45e1b12ce71ad220a6f3fb7bf574024

https://github.com/llvm/llvm-project/pull/213459