[Mlir-commits] [mlir] [XeGPU] Add missing StoreMatrix and LoadMatrix entries for bmg uArch. (PR #181006)

[Sang Ik Lee]

https://github.com/silee2 created https://github.com/llvm/llvm-project/pull/181006

uArch for bmg was missing entries for StoreMatrix and LoadMatrix instruction.

>From 3a36018a885ecbdae232de584bd5d01e563d4c88 Mon Sep 17 00:00:00 2001
From: "Lee, Sang Ik" <sang.ik.lee at intel.com>
Date: Wed, 11 Feb 2026 19:58:41 +0000
Subject: [PATCH] [XeGPU] Add missing StoreMatrix and LoadMatrix entries for
 bmg uArch.

---
 mlir/include/mlir/Dialect/XeGPU/uArch/IntelGpuXe2.h | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/mlir/include/mlir/Dialect/XeGPU/uArch/IntelGpuXe2.h b/mlir/include/mlir/Dialect/XeGPU/uArch/IntelGpuXe2.h
index 0341e4248767a..c60a744d6b858 100644
--- a/mlir/include/mlir/Dialect/XeGPU/uArch/IntelGpuXe2.h
+++ b/mlir/include/mlir/Dialect/XeGPU/uArch/IntelGpuXe2.h
@@ -271,9 +271,11 @@ struct BMGuArch : public Xe2Plus {
     static const Subgroup2DBlockPrefetchInstruction prefetchNdInst;
     static const SpirvStoreScatterInstruction storeScatterInst;
     static const SpirvLoadGatherInstruction loadGatherInst;
-    static const Instruction *arr[] = {&dpasInst,         &loadNdInst,
-                                       &storeNdInst,      &prefetchNdInst,
-                                       &storeScatterInst, &loadGatherInst};
+    static const StoreMatrixInstruction storeMatrixInst;
+    static const LoadMatrixInstruction loadMatrixInst;
+    static const Instruction *arr[] = {
+        &dpasInst,         &loadNdInst,     &storeNdInst,     &prefetchNdInst,
+        &storeScatterInst, &loadGatherInst, &storeMatrixInst, &loadMatrixInst};
     return arr;
   }