[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