a54f4eae0e
Precursor: https://reviews.llvm.org/D110200 Removed redundant ops from the standard dialect that were moved to the `arith` or `math` dialects. Renamed all instances of operations in the codebase and in tests. Reviewed By: rriddle, jpienaar Differential Revision: https://reviews.llvm.org/D110797
52 lines
1.3 KiB
MLIR
52 lines
1.3 KiB
MLIR
// RUN: mlir-opt -convert-memref-to-llvm %s | FileCheck %s
|
|
|
|
// CHECK-LABEL: @empty
|
|
func @empty() {
|
|
// CHECK: llvm.intr.stacksave
|
|
// CHECK: llvm.br
|
|
memref.alloca_scope {
|
|
memref.alloca_scope.return
|
|
}
|
|
// CHECK: llvm.intr.stackrestore
|
|
// CHECK: llvm.br
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @returns_nothing
|
|
func @returns_nothing(%b: f32) {
|
|
%a = arith.constant 10.0 : f32
|
|
// CHECK: llvm.intr.stacksave
|
|
memref.alloca_scope {
|
|
%c = arith.addf %a, %b : f32
|
|
memref.alloca_scope.return
|
|
}
|
|
// CHECK: llvm.intr.stackrestore
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @returns_one_value
|
|
func @returns_one_value(%b: f32) -> f32 {
|
|
%a = arith.constant 10.0 : f32
|
|
// CHECK: llvm.intr.stacksave
|
|
%result = memref.alloca_scope -> f32 {
|
|
%c = arith.addf %a, %b : f32
|
|
memref.alloca_scope.return %c: f32
|
|
}
|
|
// CHECK: llvm.intr.stackrestore
|
|
return %result : f32
|
|
}
|
|
|
|
// CHECK-LABEL: @returns_multiple_values
|
|
func @returns_multiple_values(%b: f32) -> f32 {
|
|
%a = arith.constant 10.0 : f32
|
|
// CHECK: llvm.intr.stacksave
|
|
%result1, %result2 = memref.alloca_scope -> (f32, f32) {
|
|
%c = arith.addf %a, %b : f32
|
|
%d = arith.subf %a, %b : f32
|
|
memref.alloca_scope.return %c, %d: f32, f32
|
|
}
|
|
// CHECK: llvm.intr.stackrestore
|
|
%result = arith.addf %result1, %result2 : f32
|
|
return %result : f32
|
|
}
|