Skip to content

Commit dca74f7

Browse files
authored
[mlir][memref] Revert #140730 (#141406)
Reverts #140730 - that turned out not to be an NFC as we originally thought. See the attached test for an example. Many thanks to @Garra1980 for reporting! Note, without this change, the newly added test would be incorrectly converted to: ```mlir func.func @view_memref_as_rank0(%arg0: index, %arg1: memref<2xi8>) { %0 = llvm.mlir.poison : !llvm.struct<(ptr, ptr, i64)> return } ```
1 parent 7511107 commit dca74f7

File tree

2 files changed

+31
-6
lines changed

2 files changed

+31
-6
lines changed

mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1721,10 +1721,6 @@ struct ViewOpLowering : public ConvertOpToLLVMPattern<memref::ViewOp> {
17211721
MemRefDescriptor sourceMemRef(adaptor.getSource());
17221722
auto targetMemRef = MemRefDescriptor::poison(rewriter, loc, targetDescTy);
17231723

1724-
// Early exit for 0-D corner case.
1725-
if (viewMemRefType.getRank() == 0)
1726-
return rewriter.replaceOp(viewOp, {targetMemRef}), success();
1727-
17281724
// Field 1: Copy the allocated pointer, used for malloc/free.
17291725
Value allocatedPtr = sourceMemRef.allocatedPtr(rewriter, loc);
17301726
auto srcMemRefType = cast<MemRefType>(viewOp.getSource().getType());
@@ -1747,6 +1743,10 @@ struct ViewOpLowering : public ConvertOpToLLVMPattern<memref::ViewOp> {
17471743
rewriter, loc,
17481744
createIndexAttrConstant(rewriter, loc, indexType, offset));
17491745

1746+
// Early exit for 0-D corner case.
1747+
if (viewMemRefType.getRank() == 0)
1748+
return rewriter.replaceOp(viewOp, {targetMemRef}), success();
1749+
17501750
// Fields 4 and 5: Update sizes and strides.
17511751
Value stride = nullptr, nextSize = nullptr;
17521752
for (int i = viewMemRefType.getRank() - 1; i >= 0; --i) {

mlir/test/Conversion/MemRefToLLVM/memref-to-llvm.mlir

Lines changed: 27 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,13 @@
1-
// RUN: mlir-opt -finalize-memref-to-llvm %s -split-input-file | FileCheck %s
1+
// RUN: mlir-opt -finalize-memref-to-llvm %s -split-input-file | FileCheck --check-prefixes=ALL,CHECK %s
22
// RUN: mlir-opt -finalize-memref-to-llvm='index-bitwidth=32' %s -split-input-file | FileCheck --check-prefix=CHECK32 %s
33

44
// Same below, but using the `ConvertToLLVMPatternInterface` entry point
55
// and the generic `convert-to-llvm` pass. This produces slightly different IR
66
// because the conversion target is set up differently.
7-
// RUN: mlir-opt --convert-to-llvm="filter-dialects=memref" --split-input-file %s | FileCheck --check-prefix=CHECK-INTERFACE %s
7+
// RUN: mlir-opt --convert-to-llvm="filter-dialects=memref" --split-input-file %s | FileCheck --check-prefixes=ALL,CHECK-INTERFACE %s
8+
9+
// TODO: In some (all?) cases, CHECK and CHECK-INTERFACE outputs are identical.
10+
// Use a common prefix instead (e.g. ALL).
811

912
// CHECK-LABEL: func @view(
1013
// CHECK: %[[ARG0F:.*]]: index, %[[ARG1F:.*]]: index, %[[ARG2F:.*]]: index
@@ -132,6 +135,28 @@ func.func @view_empty_memref(%offset: index, %mem: memref<0xi8>) {
132135

133136
// -----
134137

138+
// ALL-LABEL: func.func @view_memref_as_rank0(
139+
// ALL-SAME: %[[OFFSET:.*]]: index,
140+
// ALL-SAME: %[[MEM:.*]]: memref<2xi8>) {
141+
func.func @view_memref_as_rank0(%offset: index, %mem: memref<2xi8>) {
142+
143+
// ALL: builtin.unrealized_conversion_cast %[[OFFSET]] : index to i64
144+
// ALL: builtin.unrealized_conversion_cast %[[MEM]] : memref<2xi8> to !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>
145+
// ALL: llvm.mlir.poison : !llvm.struct<(ptr, ptr, i64)>
146+
// ALL: llvm.extractvalue %{{.*}}[0] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>
147+
// ALL: llvm.insertvalue %{{.*}}, %{{.*}}[0] : !llvm.struct<(ptr, ptr, i64)>
148+
// ALL: llvm.extractvalue %{{.*}}[1] : !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>
149+
// ALL: llvm.getelementptr %{{.*}}[%{{.*}}] : (!llvm.ptr, i64) -> !llvm.ptr, i8
150+
// ALL: llvm.insertvalue %{{.*}}, %{{.*}}[1] : !llvm.struct<(ptr, ptr, i64)>
151+
// ALL: llvm.mlir.constant(0 : index) : i64
152+
// ALL: llvm.insertvalue %{{.*}}, %{{.*}}[2] : !llvm.struct<(ptr, ptr, i64)>
153+
%memref_view_bf16 = memref.view %mem[%offset][] : memref<2xi8> to memref<bf16>
154+
155+
return
156+
}
157+
158+
// -----
159+
135160
// Subviews needs to be expanded outside of the memref-to-llvm pass.
136161
// CHECK-LABEL: func @subview(
137162
// CHECK: %[[MEMREF:.*]]: memref<{{.*}}>,

0 commit comments

Comments
 (0)