34a65980d7
Follow up to the discussion from #75258, and serves as an alternate solution for #74670. Set the location to Unknown for deduplicated / moved / materialized constants by OperationFolder. This makes sure that the folded constants don't end up with an arbitrary location of one of the original ops that became it, and that hoisted ops don't confuse the stepping order.
43 lines
1.5 KiB
MLIR
43 lines
1.5 KiB
MLIR
// RUN: mlir-opt %s -split-input-file -test-constant-fold -mlir-print-debuginfo | FileCheck %s
|
|
|
|
// CHECK-LABEL: func @fold_and_merge
|
|
func.func @fold_and_merge() -> (i32, i32) {
|
|
// CHECK-NEXT: [[C:%.+]] = arith.constant 6 : i32 loc(#[[UnknownLoc:.*]])
|
|
%0 = arith.constant 1 : i32 loc("fold_and_merge":0:0)
|
|
%1 = arith.constant 5 : i32 loc("fold_and_merge":1:0)
|
|
%2 = arith.addi %0, %1 : i32 loc("fold_and_merge":2:0)
|
|
|
|
%3 = arith.constant 6 : i32 loc("fold_and_merge":3:0)
|
|
|
|
return %2, %3: i32, i32
|
|
}
|
|
// CHECK: #[[UnknownLoc]] = loc(unknown)
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @materialize_different_dialect
|
|
func.func @materialize_different_dialect() -> (f32, f32) {
|
|
// CHECK: arith.constant 1.{{0*}}e+00 : f32 loc(#[[UnknownLoc:.*]])
|
|
%0 = arith.constant -1.0 : f32 loc("materialize_different_dialect":0:0)
|
|
%1 = math.absf %0 : f32 loc("materialize_different_dialect":1:0)
|
|
%2 = arith.constant 1.0 : f32 loc("materialize_different_dialect":2:0)
|
|
|
|
return %1, %2: f32, f32
|
|
}
|
|
// CHECK: #[[UnknownLoc]] = loc(unknown)
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @materialize_in_front
|
|
func.func @materialize_in_front(%arg0: memref<8xi32>) {
|
|
// CHECK-NEXT: arith.constant 6 : i32 loc(#[[UnknownLoc:.*]])
|
|
affine.for %arg1 = 0 to 8 {
|
|
%1 = arith.constant 1 : i32 loc("materialize_in_front":0:0)
|
|
%2 = arith.constant 5 : i32 loc("materialize_in_front":1:0)
|
|
%3 = arith.addi %1, %2 : i32 loc("materialize_in_front":2:0)
|
|
memref.store %3, %arg0[%arg1] : memref<8xi32>
|
|
}
|
|
return
|
|
} loc("materialize_in_front":3:0)
|
|
// CHECK: #[[UnknownLoc]] = loc(unknown)
|