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