[Handshake] Make sure things are integers

llvm · Nov 21, 2024 · f4d4ae9 · f4d4ae9
1 parent b538893
commit f4d4ae9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/integration_test/Dialect/Handshake/multiple_loops/multiple_loops.py b/integration_test/Dialect/Handshake/multiple_loops/multiple_loops.py
@@ -26,7 +26,7 @@ async def sendMultiple(dut):
                                                        ["out0", "out1", "out2"])
 
   N = 10
-  res0 = [i * (i + 1) / 2 for i in range(N)]
+  res0 = [i * (i + 1) // 2 for i in range(N)]
   res1 = [math.factorial(i) for i in range(N)]
 
   out0Check = cocotb.start_soon(out0.checkOutputs(res0))

diff --git a/integration_test/Dialect/Handshake/nested_loops/nested_loops.py b/integration_test/Dialect/Handshake/nested_loops/nested_loops.py
@@ -24,7 +24,7 @@ async def sendMultiple(dut):
 
   N = 10
   # sum_{i = 0}^n (sum_{j=0}^i i) = 1/6 * (n^3 + 3n^2 + 2n)
-  res = [(1 / 6) * (n**3 + 3 * n**2 + 2 * n) for n in range(N)]
+  res = [(n**3 + 3 * n**2 + 2 * n) // 6 for n in range(N)]
 
   out0Check = cocotb.start_soon(out0.checkOutputs(res))