Dynamic shape/stride/offset support by way of memref.extract_strided_…

…metadata (#11)

third_party/cpu/lib/Xsmm/ConvertTritonToXsmm.cpp

@@ -224,14 +224,13 @@ struct DotToXsmm : public OpRewritePattern<triton::DotOp> {
 
     auto brgemmInfo = xsmm::utils::isMappableToBrgemm(rewriter, dotOp, inputs,
                                                       outputs, indexingMaps);
-    if (failed(brgemmInfo))
-      return rewriter.notifyMatchFailure(dotOp, "not mappable to XSMM");
-    if (brgemmInfo->isVnni)
-      return rewriter.notifyMatchFailure(dotOp, "VNNI support NYI");
-
-    auto xsmmFuncs = xsmm::utils::buildBrgemmCalls(
-        rewriter, dotOp, ValueRange{lhsBuf, rhsBuf, accBuf}, *brgemmInfo,
-        flags);
+    if (failed(brgemmInfo)) {
+      assert(false); // FIXME: getMemrefSource above already modified IR...
+      // return rewriter.notifyMatchFailure(dotOp, "not mappable to XSMM");
+    }
+
+    auto xsmmFuncs = xsmm::utils::buildBrgemmCalls(rewriter, dotOp, inputs,
+                                                   indexingMaps, flags);
 
     if (hoistedAcc) {
       // Hoisting already updated all uses correctly.

third_party/cpu/lib/Xsmm/ConvertVectorToXsmm.cpp

@@ -180,14 +180,14 @@ struct ContractToXsmm : public OpRewritePattern<vector::ContractionOp> {
     auto brgemmInfo =
         xsmm::utils::isMappableToBrgemm(rewriter, contractOp, inputs, outputs,
                                         contractOp.getIndexingMapsArray());
-    if (failed(brgemmInfo))
-      return rewriter.notifyMatchFailure(contractOp, "not mappable to XSMM");
-    if (brgemmInfo->isVnni)
-      return rewriter.notifyMatchFailure(contractOp, "VNNI support NYI");
+    if (failed(brgemmInfo)) {
+      assert(false); // FIXME: getMemrefSource above already modified IR...
+      // return rewriter.notifyMatchFailure(contractOp, "not mappable to XSMM");
+    }
 
     auto xsmmFuncs = xsmm::utils::buildBrgemmCalls(
-        rewriter, contractOp, ValueRange{lhsBuf, rhsBuf, accBuf}, *brgemmInfo,
-        flags);
+        rewriter, contractOp, ValueRange{lhsBuf, rhsBuf, accBuf},
+        contractOp.getIndexingMapsArray(), flags);
 
     if (hoistedAcc) {
       // Hoisting already updated all uses correctly.