From 7dd5e612f40570b8c3008a8aeee90a7611949c5c Mon Sep 17 00:00:00 2001
From: sgauthamr2001 <sgauthamr2001@gmail.com>
Date: Sat, 4 Jan 2025 23:51:08 -0800
Subject: [PATCH] Use integer division.
---
src/interstellar/cost_model.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/interstellar/cost_model.py b/src/interstellar/cost_model.py
index 5599a97..42e3afd 100644
--- a/src/interstellar/cost_model.py
+++ b/src/interstellar/cost_model.py
@@ -103,7 +103,7 @@ def get_if_access(level, point, layer, mac_capacity=1):
"""
if level == 0 and mac_capacity == 0:
- return layer.wfil * layer.hfil * layer.nofm / (layer.wstd * layer.hstd)
+ return layer.wfil * layer.hfil * layer.nofm // (layer.wstd * layer.hstd)
ex_order_index = min(
point.loop_orders[le.OX][level],
@@ -126,7 +126,7 @@ def get_if_access(level, point, layer, mac_capacity=1):
oc_par = reduce(mul, point.loop_partitionings[le.OC][level:], 1)
return (
- fx_acc * fy_acc * oc_acc * fx_par * fy_par * oc_par / (layer.wstd * layer.hstd)
+ fx_acc * fy_acc * oc_acc * fx_par * fy_par * oc_par // (layer.wstd * layer.hstd)
)
--
GitLab