From 838a20c91d6cd786090569a02e8769eb525f245e Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Sun, 3 Dec 2023 20:19:34 +0100
Subject: [PATCH] Fix code for DEM
Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
src/pairs/ir/apply.py | 2 +-
src/pairs/ir/device.py | 4 ++--
src/pairs/ir/kernel.py | 2 +-
src/pairs/ir/properties.py | 2 +-
4 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/pairs/ir/apply.py b/src/pairs/ir/apply.py
index 1501fe2..5271b11 100644
--- a/src/pairs/ir/apply.py
+++ b/src/pairs/ir/apply.py
@@ -54,7 +54,7 @@ class Apply(Lowerable):
return self._build_expression_with_index(expr, index)[0]
# TODO: This method should comprise all operators and dynamic data types, it would also be
- # better to provide a better way to implement it such as a Mutator or Visitor
+ # better to provide a way to implement it with a Mutator or Visitor
def _build_expression_with_index(self, expr, index):
if isinstance(expr, (ScalarOp, VectorOp)):
new_lhs, changed_lhs = self._build_expression_with_index(expr.lhs, index)
diff --git a/src/pairs/ir/device.py b/src/pairs/ir/device.py
index 5082acf..c52613d 100644
--- a/src/pairs/ir/device.py
+++ b/src/pairs/ir/device.py
@@ -72,7 +72,7 @@ class CopyProperty(ASTNode):
class CopyContactProperty(ASTNode):
- def __init__(self, sim, prop, ctx, write):
+ def __init__(self, sim, prop, ctx, action):
super().__init__(sim)
self._contact_prop = prop
self._ctx = ctx
@@ -89,7 +89,7 @@ class CopyContactProperty(ASTNode):
return self._action
def children(self):
- return [self._prop]
+ return [self._contact_prop]
class CopyVar(ASTNode):
diff --git a/src/pairs/ir/kernel.py b/src/pairs/ir/kernel.py
index e5e19c9..04def29 100644
--- a/src/pairs/ir/kernel.py
+++ b/src/pairs/ir/kernel.py
@@ -116,7 +116,7 @@ class Kernel(ASTNode):
new_op = 'w' if write else 'r'
for contact_prop in contact_prop_list:
- assert isinstance(cp, ContactProperty), \
+ assert isinstance(contact_prop, ContactProperty), \
"Kernel.add_contact_property(): Element is not of type ContactProperty."
action = Actions.NoAction if contact_prop not in self._contact_properties else \
diff --git a/src/pairs/ir/properties.py b/src/pairs/ir/properties.py
index f841199..f10f942 100644
--- a/src/pairs/ir/properties.py
+++ b/src/pairs/ir/properties.py
@@ -258,7 +258,7 @@ class ContactProperty(ASTNode):
def copy_size(self):
return ScalarOp.inline(self.sim.nlocal *
self.sim.neighbor_capacity *
- Types.number_of_elements(self.sim, self.prop_type) *
+ Types.number_of_elements(self.sim, self.contact_prop_type) *
Sizeof(self.sim, self.type()))
def sizes(self):
--
GitLab