From 2eea3159e46d92774d6253966316cb3dfb4a6928 Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Fri, 1 Apr 2022 01:59:34 +0200
Subject: [PATCH] Fix BinOps passed as parameters in kernels
Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
src/pairs/analysis/devices.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pairs/analysis/devices.py b/src/pairs/analysis/devices.py
index 4558953..0e61269 100644
--- a/src/pairs/analysis/devices.py
+++ b/src/pairs/analysis/devices.py
@@ -48,7 +48,7 @@ class FetchKernelReferences(Visitor):
def visit_Decl(self, ast_node):
if isinstance(ast_node.elem, BinOp):
for k in self.kernel_stack:
- self.kernel_decls[k.kernel_id].append(ast_node)
+ self.kernel_decls[k.kernel_id].append(ast_node.elem)
def visit_BinOp(self, ast_node):
for k in self.kernel_stack:
--
GitLab