[Mlir-commits] [mlir] [mlir][nvgpu] add `nvgpu.rcp` op (PR #100965)

Guray Ozen llvmlistbot at llvm.org
Mon Jul 29 03:43:13 PDT 2024


================
@@ -802,4 +814,24 @@ def NVGPU_WarpgroupMmaInitAccumulatorOp : NVGPU_Op<"warpgroup.mma.init.accumulat
   let hasVerifier = 1;
 }
 
+def NVGPU_RcpOp : NVGPU_Op<"rcp", [Pure,
+                                   SameOperandsAndResultShape,
+                                   PredOpTrait<"in and out have same element type",
+                                            TCresVTEtIsSameAsOp<0, 0>>]> {
+  let summary = "F32 rcp calculation for vector types using nvvm.rcp.* OP";
----------------
grypp wrote:

```suggestion
  let summary = "The reciprocal calculation for vector types";
```

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


More information about the Mlir-commits mailing list