fannkuch: broken but fix some bugs

blocked on facebookincubator/cinder#115 and facebookincubator/cinder#116
utahplt · Jul 21, 2023 · 5b8f0b1 · 5b8f0b1
1 parent b1cb10b
commit 5b8f0b1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/Benchmark/fannkuch/advanced/main.py b/Benchmark/fannkuch/advanced/main.py
@@ -11,18 +11,17 @@
 import sys
 
 DEFAULT_ARG = 9
-ArrayI64 = Array[int64]
 
 
 def fannkuch(nb: int) -> int:
     n: int64 = int64(nb)
-    count: ArrayI64 = ArrayI64(range(1, nb + 1))
+    count: Array[int64] = Array[int64](nb)
     max_flips: int64 = 0
     m: int64 = n - 1
     r: int64 = n
-    perm1: ArrayI64 = ArrayI64(range(nb))
-    perm: ArrayI64 = ArrayI64(range(nb))
-    perm0: ArrayI64 = ArrayI64(range(nb))
+    perm1: Array[int64] = Array[int64](nb)
+    perm: Array[int64] = Array[int64](nb)
+    perm0: Array[int64] = Array[int64](nb)
 
     while 1:
         while r != 1:
@@ -64,11 +63,12 @@ def fannkuch(nb: int) -> int:
             r += 1
         else:
             return box(max_flips)
+    return 0
 
 if __name__ == "__main__":
     num_iterations = 1
     if len(sys.argv) > 1:
         num_iterations = int(sys.argv[1])
     for _ in range(num_iterations):
         res = fannkuch(DEFAULT_ARG)
-        assert res == 30
+        assert res == 30
diff --git a/Benchmark/fannkuch/shallow/main.py b/Benchmark/fannkuch/shallow/main.py
@@ -48,11 +48,12 @@ def fannkuch(n: int) -> int:
             r += 1
         else:
             return max_flips
+    return 0
 
 if __name__ == "__main__":
     num_iterations = 1
     if len(sys.argv) > 1:
         num_iterations = int(sys.argv[1])
     for _ in range(num_iterations):
         res = fannkuch(DEFAULT_ARG)
-        assert res == 30
+        assert res == 30
diff --git a/Benchmark/fannkuch/untyped/main.py b/Benchmark/fannkuch/untyped/main.py
@@ -43,6 +43,7 @@ def fannkuch(n):
             r += 1
         else:
             return max_flips
+    return 0
 
 if __name__ == "__main__":
     import sys
@@ -52,4 +53,4 @@ def fannkuch(n):
         num_iterations = int(sys.argv[1])
     for _ in range(num_iterations):
         res = fannkuch(DEFAULT_ARG)
-        assert res == 30
+        assert res == 30