update polybench to use invokeable sqrt

vmiheer · Jan 30, 2021 · 7117bb0 · 7117bb0
1 parent acf96be
commit 7117bb0
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 10 deletions.
diff --git a/benchmarks/polybench/linear-algebra-cholesky.fuse b/benchmarks/polybench/linear-algebra-cholesky.fuse
@@ -1,6 +1,7 @@
 import c++("math.h")
-       vivado("math.h") {
-  def sqrt(x: ubit<32>): ubit<32>;
+       vivado("math.h")
+       futil("primitives/bitnum/math.futil") {
+  def sqrt(in: ubit<32>): ubit<32>;
 }
 
 decl A_int: ubit<32>[8][8];
@@ -55,5 +56,6 @@ for (let i: ubit<4> = 0..8) {
   ---
   let A_i_i = A_int[i][i];
   ---
-  A_int[i][i] := sqrt(A_i_i);
+  let sq = sqrt(A_i_i);
+  A_int[i][i] := sq;
 }
diff --git a/benchmarks/polybench/linear-algebra-gramschmidt.fuse b/benchmarks/polybench/linear-algebra-gramschmidt.fuse
@@ -1,6 +1,7 @@
 import c++("math.h")
-       vivado("math.h") {
-  def sqrt(x: ubit<32>): ubit<32>;
+       vivado("math.h")
+       futil("primitives/bitnum/math.futil") {
+  def sqrt(in: ubit<32>): ubit<32>;
 }
 
 decl A_int: ubit<32>[8][12];
@@ -39,7 +40,8 @@ for (let k: ubit<4> = 0..12) {
     nrm += A_i_k * A_i_k;
   }
 
-  R[k][k] := sqrt(nrm);
+  let sq = sqrt(nrm);
+  R[k][k] := sq;
   ---
   for (let i: ubit<4> = 0..8) {
     Q[i][k] := A[i][k] / R[k][k];

diff --git a/benchmarks/unrolled/linear-algebra-gramschmidt.fuse b/benchmarks/unrolled/linear-algebra-gramschmidt.fuse
@@ -1,6 +1,7 @@
-import vivado("math.h")
-       cpp("math.h") {
-  def sqrt(x: ubit<32>): ubit<32>;
+import c++("math.h")
+       vivado("math.h")
+       futil("primitives/bitnum/math.futil") {
+  def sqrt(in: ubit<32>): ubit<32>;
 }
 
 decl A_int: ubit<32>[8][12];
@@ -39,7 +40,8 @@ for (let k: ubit<4> = 0..12) {
     nrm += A_i_k * A_i_k;
   }
 
-  R[k][k] := sqrt(nrm);
+  let sq = sqrt(nrm);
+  R[k][k] := sq;
   ---
   for (let i: ubit<4> = 0..8) unroll 4 {
     Q[i][k] := A[i][k] / R[k][k];