aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/target/layout/AOS.py
diff options
context:
space:
mode:
Diffstat (limited to 'boltzgen/kernel/target/layout/AOS.py')
-rw-r--r--boltzgen/kernel/target/layout/AOS.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/boltzgen/kernel/target/layout/AOS.py b/boltzgen/kernel/target/layout/AOS.py
index c603869..88e5377 100644
--- a/boltzgen/kernel/target/layout/AOS.py
+++ b/boltzgen/kernel/target/layout/AOS.py
@@ -1,7 +1,7 @@
class AOS:
- def __init__(self, descriptor, cell_index, geometry):
+ def __init__(self, descriptor, index, geometry):
self.descriptor = descriptor
- self.cell_index = cell_index
+ self.index = index(geometry)
self.geometry = geometry
def cell_preshift(self, gid):
@@ -11,4 +11,4 @@ class AOS:
return i
def neighbor_offset(self, c_i):
- return self.descriptor.q * self.cell_index.neighbor(c_i)
+ return self.descriptor.q * self.index.neighbor(c_i)