From 2bfb17a20293d20102a9914e5e5386716e7faa62 Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Fri, 4 Nov 2022 20:35:15 +0100
Subject: [PATCH] Check device configuration before launching kernel
Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
src/pairs/code_gen/cgen.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/pairs/code_gen/cgen.py b/src/pairs/code_gen/cgen.py
index 516479b..270a8b1 100644
--- a/src/pairs/code_gen/cgen.py
+++ b/src/pairs/code_gen/cgen.py
@@ -392,7 +392,11 @@ class CGen:
threads_per_block = self.generate_expression(ast_node.threads_per_block)
nblocks = self.generate_expression(ast_node.nblocks)
+ self.print(f"if({nblocks} > 0 && {threads_per_block} > 0) {{")
+ self.print.add_indent(4)
self.print(f"{kernel.name}<<<{nblocks}, {threads_per_block}>>>({kernel_params});")
+ self.print.add_indent(-4)
+ self.print("}")
if isinstance(ast_node, ModuleCall):
module = ast_node.module
--
GitLab