From 79be6ee25c1dad71653b5f63b3c01f1f292fa961 Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Fri, 4 Nov 2022 03:11:12 +0100
Subject: [PATCH] Use device pointer on module parameters for variables
Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
src/pairs/code_gen/cgen.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pairs/code_gen/cgen.py b/src/pairs/code_gen/cgen.py
index 5d3c9db..516479b 100644
--- a/src/pairs/code_gen/cgen.py
+++ b/src/pairs/code_gen/cgen.py
@@ -404,7 +404,7 @@ class CGen:
module_params += decl if len(module_params) <= 0 else f", {decl}"
for var in module.write_variables():
- decl = f"&{var.name()}"
+ decl = f"rv_{var.name()}.getDevicePointer()" if device_cond and var.device_flag else f"&{var.name()}"
module_params += decl if len(module_params) <= 0 else f", {decl}"
for array in module.arrays():
--
GitLab