Support math.erf operation

UIUC-ChenLab · Jan 9, 2024 · 02273bb · 02273bb
1 parent 4bf297f
commit 02273bb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/include/scalehls/Utils/Visitor.h b/include/scalehls/Utils/Visitor.h
@@ -51,7 +51,7 @@ class HLSVisitorBase {
             // Unary expressions.
             math::AbsIOp, math::AbsFOp, math::CeilOp, math::CosOp, math::SinOp,
             math::TanhOp, math::SqrtOp, math::RsqrtOp, math::ExpOp,
-            math::Exp2Op, math::LogOp, math::Log2Op, math::Log10Op,
+            math::Exp2Op, math::LogOp, math::Log2Op, math::Log10Op, math::ErfOp,
             arith::NegFOp,
 
             // Float binary expressions.
@@ -158,6 +158,7 @@ class HLSVisitorBase {
   HANDLE(math::LogOp);
   HANDLE(math::Log2Op);
   HANDLE(math::Log10Op);
+  HANDLE(math::ErfOp);
   HANDLE(arith::NegFOp);
 
   // Float binary expressions.

diff --git a/lib/Translation/EmitHLSCpp.cpp b/lib/Translation/EmitHLSCpp.cpp
@@ -573,6 +573,7 @@ class ExprVisitor : public HLSVisitorBase<ExprVisitor, bool> {
   bool visitOp(math::Log10Op op) {
     return emitter.emitUnary(op, "log10"), true;
   }
+  bool visitOp(math::ErfOp op) { return emitter.emitUnary(op, "erf"), true; }
   bool visitOp(arith::NegFOp op) { return emitter.emitUnary(op, "-"), true; }
 
   /// Float binary expressions.
@@ -1877,6 +1878,7 @@ void ModuleEmitter::emitModule(ModuleOp module) {
 //===----------------------------------------------------------------------===//
 
 #include <algorithm>
+#include <cmath>
 #include <ap_axi_sdata.h>
 #include <ap_fixed.h>
 #include <ap_int.h>