forked from OSchip/llvm-project
22 lines
838 B
MLIR
22 lines
838 B
MLIR
// RUN: mlir-opt %s -test-print-topological-sort 2>&1 | FileCheck %s
|
|
|
|
// CHECK-LABEL: Testing : region
|
|
// CHECK: arith.addi {{.*}} : index
|
|
// CHECK-NEXT: scf.for
|
|
// CHECK: } {__test_sort_original_idx__ = 2 : i64}
|
|
// CHECK-NEXT: arith.addi {{.*}} : i32
|
|
// CHECK-NEXT: arith.subi {{.*}} : i32
|
|
func @region(
|
|
%arg0 : index, %arg1 : index, %arg2 : index, %arg3 : index,
|
|
%arg4 : i32, %arg5 : i32, %arg6 : i32,
|
|
%buffer : memref<i32>) {
|
|
%0 = arith.addi %arg4, %arg5 {__test_sort_original_idx__ = 0} : i32
|
|
%idx = arith.addi %arg0, %arg1 {__test_sort_original_idx__ = 3} : index
|
|
scf.for %arg7 = %idx to %arg2 step %arg3 {
|
|
%2 = arith.addi %0, %arg5 : i32
|
|
%3 = arith.subi %2, %arg6 {__test_sort_original_idx__ = 1} : i32
|
|
memref.store %3, %buffer[] : memref<i32>
|
|
} {__test_sort_original_idx__ = 2}
|
|
return
|
|
}
|