[Mlir-commits] [mlir] [MLIR][XeGPU] Add support for elementwise ops in Wg to Sg distribute pass [1/N] (PR #142797)

[Nishant Patel]

================
@@ -314,6 +317,90 @@ struct WgToSgPrefetchNdOp : public OpConversionPattern<xegpu::PrefetchNdOp> {
   }
 };
 
+// This pattern transforms elementwise ops (unary/binary) in math/arith dialect
+template <typename Op>
+struct WgToSgElementwiseOp : public OpConversionPattern<Op> {
----------------
nbpatel wrote:

I tried using it, but I was not able to use OneToNOpAdator (which I need to get the updated operands) without templating it. Do you guys have any pointers which I can refer to? 

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