[Mlir-commits] [mlir] [NvGpu Dialect] add rcp approxe op (PR #100965)

Guray Ozen llvmlistbot at llvm.org
Sun Jul 28 23:27:08 PDT 2024


================
@@ -802,4 +803,16 @@ def NVGPU_WarpgroupMmaInitAccumulatorOp : NVGPU_Op<"warpgroup.mma.init.accumulat
   let hasVerifier = 1;
 }
 
+def NVGPU_RcpApproxOp : NVGPU_Op<"rcp_approx", [
+  Pure, SameOperandsAndResultType
+]> {
+  let summary = "A wrapper of nvvm rcp.approx.ftz.f";
----------------
grypp wrote:

Nit: It's beyond a wrapper. The lowering transforms multi dimensional vector, and generates nvvm OP for each of them. Maybe let's say something like
F32 rcp calculation for vector types using `nvvm.rcp.*` OP

https://github.com/llvm/llvm-project/pull/100965


More information about the Mlir-commits mailing list