onnx-mlir/test/mlir/krnl/sqrt.mlir

25 lines
961 B
MLIR

// RUN: onnf-opt --shape-inference --lower-all-llvm %s -split-input-file | FileCheck %s
module {
func @test_sqrt_32(%arg0 : f32) -> f32 {
%0 = "krnl.sqrt"(%arg0) : (f32) -> f32
"std.return"(%0) : (f32) -> ()
// CHECK: llvm.func @llvm.sqrt.f32(!llvm.float) -> !llvm.float
// CHECK-NEXT: llvm.func @test_sqrt_32(%arg0: !llvm.float) -> !llvm.float {
// CHECK-NEXT: [[RES:%.+]] = llvm.call @llvm.sqrt.f32(%arg0) : (!llvm.float) -> !llvm.float
// CHECK-NEXT: llvm.return [[RES]] : !llvm.float
}
}
module{
func @test_sqrt_64(%arg0 : f64) -> f64 {
%0 = "krnl.sqrt"(%arg0) : (f64) -> f64
"std.return"(%0) : (f64) -> ()
// CHECK: llvm.func @llvm.sqrt.f64(!llvm.double) -> !llvm.double
// CHECK-NEXT: llvm.func @test_sqrt_64(%arg0: !llvm.double) -> !llvm.double {
// CHECK-NEXT: [[RES:%.+]] = llvm.call @llvm.sqrt.f64(%arg0) : (!llvm.double) -> !llvm.double
// CHECK-NEXT: llvm.return [[RES]] : !llvm.double
}
}