From c81353e5b8f384155ab01f3eda574c2ee44fd69a Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Thu, 22 Aug 2019 11:31:34 +0530
Subject: [PATCH] handle return tuple correctly
---
loopy/check.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/loopy/check.py b/loopy/check.py
index aeec19c78..ad7717226 100644
--- a/loopy/check.py
+++ b/loopy/check.py
@@ -30,8 +30,8 @@ import islpy as isl
from loopy.symbolic import WalkMapper
from loopy.diagnostic import LoopyError, WriteRaceConditionWarning, warn_with_kernel
from loopy.type_inference import TypeInferenceMapper
-from loopy.kernel.instruction import (MultiAssignmentBase, CInstruction,
- _DataObliviousInstruction)
+from loopy.kernel.instruction import (MultiAssignmentBase, CallInstruction,
+ CInstruction, _DataObliviousInstruction)
import logging
logger = logging.getLogger(__name__)
@@ -83,7 +83,8 @@ def check_for_integer_subscript_indices(kernel):
idx_int_checker = SubscriptIndicesIsIntChecker(kernel)
for insn in kernel.instructions:
if isinstance(insn, MultiAssignmentBase):
- idx_int_checker(insn.expression)
+ idx_int_checker(insn.expression, return_tuple=isinstance(insn,
+ CallInstruction), return_dtype_set=True)
[idx_int_checker(assignee) for assignee in insn.assignees]
elif isinstance(insn, (CInstruction, _DataObliviousInstruction)):
pass
--
GitLab