This is an archive of the discontinued LLVM Phabricator instance.

Differential D153347

[mlir][Interfaces][NFC] TilingInterface: Add test case for linalg.copy on memrefs
ClosedPublic

Authored by springerm on Jun 20 2023, 7:02 AM.

Details

Reviewers
nicolasvasilache
Commits
rG226845998509: [mlir][Interfaces] TilingInterface: Add test case for linalg.copy on memrefs

Diff Detail

Event Timeline

springerm created this revision.Jun 20 2023, 7:02 AM
Herald added a project: Restricted Project. · View Herald TranscriptJun 20 2023, 7:02 AM
Herald added subscribers: bviyer, Moerafaat, zero9178 and 22 others. · View Herald Transcript
springerm requested review of this revision.Jun 20 2023, 7:02 AM
Herald added a project: Restricted Project. · View Herald TranscriptJun 20 2023, 7:02 AM
Herald added subscribers: limo1996, stephenneuendorffer. · View Herald Transcript
springerm retitled this revision from [mlir][Interfaces] TilingInterface: Add test case for linalg.copy on memrefs to [mlir][Interfaces][NFC] TilingInterface: Add test case for linalg.copy on memrefs.Jun 20 2023, 7:02 AM
Harbormaster completed remote builds in B240003: Diff 532912.Jun 20 2023, 7:45 AM
nicolasvasilache accepted this revision.Jun 20 2023, 8:57 AM
This revision is now accepted and ready to land.Jun 20 2023, 8:57 AM
Closed by commit rG226845998509: [mlir][Interfaces] TilingInterface: Add test case for linalg.copy on memrefs (authored by springerm). · Explain WhyJun 20 2023, 11:57 PM
This revision was automatically updated to reflect the committed changes.
springerm added a commit: rG226845998509: [mlir][Interfaces] TilingInterface: Add test case for linalg.copy on memrefs.

Revision Contents

PathSize
mlir/
test/
Interfaces/
TilingInterface/
14 lines
lib/
Interfaces/
TilingInterface/
3 lines

Diff 533152

mlir/test/Interfaces/TilingInterface/tile-using-interface.mlir

Show First 20 Lines • Show All 267 Lines • ▼ Show 20 Lines
// CHECK-SAME: ins(%[[LHS_TILE]], %[[RHS_TILE]] : // CHECK-SAME: ins(%[[LHS_TILE]], %[[RHS_TILE]] :
// CHECK-SAME: outs(%[[INIT_TILE]] : // CHECK-SAME: outs(%[[INIT_TILE]] :
// CHECK: %[[UPDATE:.+]] = tensor.insert_slice %[[GEMM_TILE]] into %[[INIT2]] // CHECK: %[[UPDATE:.+]] = tensor.insert_slice %[[GEMM_TILE]] into %[[INIT2]]
// CHECK-SAME: [%[[IV2]], %[[IV0]]] [%[[TS_M]], %[[TS_N]]] [1, 1] // CHECK-SAME: [%[[IV2]], %[[IV0]]] [%[[TS_M]], %[[TS_N]]] [1, 1]
// CHECK: scf.yield %[[UPDATE]] // CHECK: scf.yield %[[UPDATE]]
// CHECK: scf.yield %[[INNER2]] // CHECK: scf.yield %[[INNER2]]
// CHECK: scf.yield %[[INNER1]] // CHECK: scf.yield %[[INNER1]]
// CHECK: return %[[OUTER]] // CHECK: return %[[OUTER]]
// -----
// CHECK-LABEL: func @linalg_copy_matmul(
// CHECK: scf.for
// CHECK: scf.for
// CHECK: memref.subview
// CHECK: memref.subview
// CHECK: linalg.copy
func.func @linalg_copy_matmul(%a: memref<?x?xf32>, %b: memref<?x?xf32>) {
linalg.copy {__internal_linalg_transform__ = "simple_copy_memref"}
ins(%a : memref<?x?xf32>) outs(%b : memref<?x?xf32>)
return
}

mlir/test/lib/Interfaces/TilingInterface/TestTilingInterface.cpp

Show First 20 Lines • Show All 501 Lines • ▼ Show 20 Lines if (testTiling) {
addPatternForTiling(context, patterns, "gemm_interchange", {10, 20, 30}, addPatternForTiling(context, patterns, "gemm_interchange", {10, 20, 30},
{1, 2, 0}); {1, 2, 0});
// 7. Tiling for 2D pad tensor operations. // 7. Tiling for 2D pad tensor operations.
addPatternForTiling(context, patterns, "pad_2dtiling", {2, 3}); addPatternForTiling(context, patterns, "pad_2dtiling", {2, 3});
// 8. Tiling inner dimension of 2d pad tensor operations. // 8. Tiling inner dimension of 2d pad tensor operations.
addPatternForTiling(context, patterns, "pad_inner_tiling", {0, 3}); addPatternForTiling(context, patterns, "pad_inner_tiling", {0, 3});
// 9. Tiling inner dimension of 2d pad tensor operations. // 9. Tiling inner dimension of 2d pad tensor operations.
addPatternForTiling(context, patterns, "pad_outer_tiling", {2, 3}); addPatternForTiling(context, patterns, "pad_outer_tiling", {2, 3});
// 10. Tiling M and N dims of `linalg.copy` on memrefs.
addPatternForTiling(context, patterns, "simple_copy_memref", {10, 20});
return; return;
} }
if (testTileConsumerAndFuseProducer) { if (testTileConsumerAndFuseProducer) {
// 1. Tile and fuse of gemm with fill producer and bias-add consumer. // 1. Tile and fuse of gemm with fill producer and bias-add consumer.
addPatternForTileAndFuse(context, patterns, "fusion", {10, 20}); addPatternForTileAndFuse(context, patterns, "fusion", {10, 20});
// 2. Tile and fuse sequence of GEMMs, by fusing only along M. // 2. Tile and fuse sequence of GEMMs, by fusing only along M.
addPatternForTileAndFuse(context, patterns, "gemm_fusion", {10}); addPatternForTileAndFuse(context, patterns, "gemm_fusion", {10});
// 3. Tile and fuse gemm with consumer + interchange of tiled loops. // 3. Tile and fuse gemm with consumer + interchange of tiled loops.
▲ Show 20 Lines • Show All 42 Lines • Show Last 20 Lines