[clang] [mlir] Add ContractionOpInterface utility functions for vector matrix multiplication (PR #68945)

2023-10-16 Thread Jerry Wu via cfe-commits
@@ -240,4 +240,134 @@ TEST(isRowMajorBatchMatmul, FirstInputSwapped) { EXPECT_THAT(maps, Not(Truly(isRowMajorBatchMatmul))); } +TEST(isVecmat, Simple) { + MLIRContext context; + + AffineExpr k, n; + bindDims(&context, k, n); + auto mapA = AffineMapAttr::get(AffineMap::ge

[clang] [mlir] Add ContractionOpInterface utility functions for vector matrix multiplication (PR #68945)

2023-10-16 Thread Jerry Wu via cfe-commits
@@ -240,4 +240,134 @@ TEST(isRowMajorBatchMatmul, FirstInputSwapped) { EXPECT_THAT(maps, Not(Truly(isRowMajorBatchMatmul))); } +TEST(isVecmat, Simple) { + MLIRContext context; + + AffineExpr k, n; + bindDims(&context, k, n); + auto mapA = AffineMapAttr::get(AffineMap::ge

[clang] [mlir] Add ContractionOpInterface utility functions for vector matrix multiplication (PR #68945)

2023-10-16 Thread Jerry Wu via cfe-commits
@@ -240,4 +240,134 @@ TEST(isRowMajorBatchMatmul, FirstInputSwapped) { EXPECT_THAT(maps, Not(Truly(isRowMajorBatchMatmul))); } +TEST(isVecmat, Simple) { + MLIRContext context; + + AffineExpr k, n; + bindDims(&context, k, n); + auto mapA = AffineMapAttr::get(AffineMap::ge