From 3bc599c3382d03adf25bcc3a25b30a8e8f42c2b9 Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Fri, 4 Nov 2022 02:58:47 +0100
Subject: [PATCH] Fix generation of runtime variable copies
Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
src/pairs/code_gen/cgen.py | 4 ++--
src/pairs/transformations/__init__.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/pairs/code_gen/cgen.py b/src/pairs/code_gen/cgen.py
index 9cdb406..5d3c9db 100644
--- a/src/pairs/code_gen/cgen.py
+++ b/src/pairs/code_gen/cgen.py
@@ -300,9 +300,9 @@ class CGen:
var_name = ast_node.variable.name()
if ast_node.context() == Contexts.Device:
- self.print(f"rv_{var_name}->copyToDevice();")
+ self.print(f"rv_{var_name}.copyToDevice();")
else:
- self.print(f"rv_{var_name}->copyToHost();")
+ self.print(f"rv_{var_name}.copyToHost();")
if isinstance(ast_node, ClearArrayFlag):
array_id = ast_node.array.id()
diff --git a/src/pairs/transformations/__init__.py b/src/pairs/transformations/__init__.py
index 00c5a35..e1cf7d3 100644
--- a/src/pairs/transformations/__init__.py
+++ b/src/pairs/transformations/__init__.py
@@ -81,8 +81,8 @@ class Transformations:
self.lift_expressions_to_owner_blocks()
self.licm()
self.modularize()
- self.add_device_copies()
self.add_device_kernels()
+ self.add_device_copies()
self.lower(True)
self.add_expression_declarations()
self.add_host_references_to_modules()
--
GitLab