Implement rcp instruction

1 files changed, 29 insertions, 0 deletions

diff --git a/ptx/src/ptx.lalrpop b/ptx/src/ptx.lalrpop
index c29d16b..a132705 100644
--- a/ptx/src/ptx.lalrpop
+++ b/ptx/src/ptx.lalrpop
@@ -35,6 +35,7 @@ match {
     ".address_size",
     ".align",
     ".and",
+    ".approx",
     ".b16",
     ".b32",
     ".b64",
@@ -134,6 +135,7 @@ match {
     "mul",
     "not",
     "or",
+    "rcp",
     "ret",
     "setp",
     "shl",
@@ -166,6 +168,7 @@ ExtendedID : &'input str = {
     "mul",
     "not",
     "or",
+    "rcp",
     "ret",
     "setp",
     "shl",
@@ -542,6 +545,7 @@ Instruction: ast::Instruction<ast::ParsedArgParams<'input>> = {
     InstSub,
     InstMin,
     InstMax,
+    InstRcp
 };
 
 // https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#data-movement-and-conversion-instructions-ld
@@ -1119,6 +1123,31 @@ OrType: ast::OrType = {
     ".b64" => ast::OrType::B64,
 }
 
+// https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#floating-point-instructions-rcp
+InstRcp: ast::Instruction<ast::ParsedArgParams<'input>> = {
+    "rcp" <rounding:RcpRoundingMode> <ftz:".ftz"?> ".f32" <a:Arg2> => {
+        let details = ast::RcpDetails {
+            rounding,
+            flush_to_zero: ftz.is_some(),
+            is_f64: false,
+        };
+        ast::Instruction::Rcp(details, a)
+    },
+    "rcp" <rn:RoundingModeFloat> ".f64" <a:Arg2> => {
+        let details = ast::RcpDetails {
+            rounding: Some(rn),
+            flush_to_zero: false,
+            is_f64: true,
+        };
+        ast::Instruction::Rcp(details, a)
+    }
+};
+
+RcpRoundingMode: Option<ast::RoundingMode> = {
+    ".approx" => None,
+    <r:RoundingModeFloat> => Some(r)
+};
+
 // https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#integer-arithmetic-instructions-sub
 // https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#floating-point-instructions-sub
 // https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#half-precision-floating-point-instructions-sub