25 lines · plain
1// RUN: mlir-opt %s -llvm-optimize-for-nvvm-target | FileCheck %s2 3// CHECK-LABEL: llvm.func @fdiv_fp164llvm.func @fdiv_fp16(%arg0 : f16, %arg1 : f16) -> f16 {5 // CHECK-DAG: %[[c0:.*]] = llvm.mlir.constant(0 : ui32) : i326 // CHECK-DAG: %[[mask:.*]] = llvm.mlir.constant(2139095040 : ui32) : i327 // CHECK-DAG: %[[lhs:.*]] = llvm.fpext %arg0 : f16 to f328 // CHECK-DAG: %[[rhs:.*]] = llvm.fpext %arg1 : f16 to f329 // CHECK-DAG: %[[rcp:.*]] = nvvm.rcp.approx.ftz.f %[[rhs]] : f3210 // CHECK-DAG: %[[approx:.*]] = llvm.fmul %[[lhs]], %[[rcp]] : f3211 // CHECK-DAG: %[[neg:.*]] = llvm.fneg %[[rhs]] : f3212 // CHECK-DAG: %[[err:.*]] = llvm.intr.fma(%[[approx]], %[[neg]], %[[lhs]]) : (f32, f32, f32) -> f3213 // CHECK-DAG: %[[refined:.*]] = llvm.intr.fma(%[[err]], %[[rcp]], %[[approx]]) : (f32, f32, f32) -> f3214 // CHECK-DAG: %[[cast:.*]] = llvm.bitcast %[[approx]] : f32 to i3215 // CHECK-DAG: %[[exp:.*]] = llvm.and %[[cast]], %[[mask]] : i3216 // CHECK-DAG: %[[is_zero:.*]] = llvm.icmp "eq" %[[exp]], %[[c0]] : i3217 // CHECK-DAG: %[[is_mask:.*]] = llvm.icmp "eq" %[[exp]], %[[mask]] : i3218 // CHECK-DAG: %[[pred:.*]] = llvm.or %[[is_zero]], %[[is_mask]] : i119 // CHECK-DAG: %[[select:.*]] = llvm.select %[[pred]], %[[approx]], %[[refined]] : i1, f3220 // CHECK-DAG: %[[result:.*]] = llvm.fptrunc %[[select]] : f32 to f1621 %result = llvm.fdiv %arg0, %arg1 : f1622 // CHECK: llvm.return %[[result]] : f1623 llvm.return %result : f1624}25