From 177851833cc9de432fe7f8eead639d5bb293d534 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 24 Feb 2016 01:26:04 -0600
Subject: [PATCH] ISPC: Don't assume a local size is present
---
loopy/target/ispc.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/loopy/target/ispc.py b/loopy/target/ispc.py
index 2c751e103..b59ed04c3 100644
--- a/loopy/target/ispc.py
+++ b/loopy/target/ispc.py
@@ -121,9 +121,12 @@ class ISPCTarget(CTarget):
from pymbolic.mapper.stringifier import PREC_COMPARISON, PREC_NONE
ccm = self.get_expression_to_code_mapper(codegen_state)
- wrapper_body.extend([
- S("assert(programCount == %s)"
- % ccm(lsize[0], PREC_COMPARISON)),
+ if lsize:
+ wrapper_body.append(
+ S("assert(programCount == %s)"
+ % ccm(lsize[0], PREC_COMPARISON)))
+
+ wrapper_body.append(
S("launch[%s] %s(%s)"
% (
", ".join(
@@ -132,7 +135,7 @@ class ISPCTarget(CTarget):
inner_name,
", ".join(arg_names)
))
- ])
+ )
wrapper_fbody = FunctionBody(
ISPCExport(
--
GitLab