[Mlir-commits] [mlir] [MLIR][XeGPU] Add lowering from transfer_read/transfer_write to load_gather/store_scatter (PR #152429)

[Jianhui Li]

================
@@ -404,3 +406,18 @@ void xegpu::doSCFStructuralTypeConversionWithTensorType(
     (void)mlir::applyPartialConversion(op, target, std::move(patterns));
   }
 }
+
+std::optional<std::string> xegpu::getXeGPUChipStr(Operation *op) {
+  auto gpuModuleOp = op->getParentOfType<mlir::gpu::GPUModuleOp>();
+  if (gpuModuleOp) {
+    auto targetAttrs = gpuModuleOp.getTargets();
+    if (targetAttrs) {
+      for (auto &attr : *targetAttrs) {
+        auto xevmAttr = llvm::dyn_cast<mlir::xevm::XeVMTargetAttr>(attr);
+        if (xevmAttr)
+          return xevmAttr.getChip().str();
+      }
+    }
+  }
+  return std::nullopt;
+}
----------------
Jianhui-Li wrote:

added

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