Add Binary Intrinsic ops to TIR Ops in C++ (apache#5900)

* Add Binary Intrinsic ops to TIR Ops in C++ * clang-format
neo-ai · Jul 2, 2020 · 30e8995 · 30e8995
1 parent bae43a6
commit 30e8995
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/include/tvm/tir/op.h b/include/tvm/tir/op.h
@@ -583,6 +583,18 @@ TVM_DECLARE_INTRIN_UNARY(acosh);
 TVM_DECLARE_INTRIN_UNARY(asinh);
 TVM_DECLARE_INTRIN_UNARY(atanh);
 
+#define TVM_DECLARE_INTRIN_BINARY(OpName)                                  \
+  inline PrimExpr OpName(PrimExpr x, PrimExpr y) {                         \
+    static const Op& op = Op::Get("tir." #OpName);                         \
+    return tir::Call(x.dtype(), op, {x, y}, tir::CallNode::PureIntrinsic); \
+  }
+
+TVM_DECLARE_INTRIN_BINARY(atan2);
+TVM_DECLARE_INTRIN_BINARY(nextafter);
+TVM_DECLARE_INTRIN_BINARY(copysign);
+TVM_DECLARE_INTRIN_BINARY(hypot);
+TVM_DECLARE_INTRIN_BINARY(ldexp);
+
 namespace tir {
 /*!
  * \brief Make a const value with certain data type.