diff --git a/examples/python/call-external.py b/examples/python/call-external.py index 68618a7ecaaaba9c9c93495a05c2313a7660b377..c13d99bd06295096c26d6e113841c853f80645fc 100644 --- a/examples/python/call-external.py +++ b/examples/python/call-external.py @@ -7,14 +7,14 @@ from loopy.target.c import CTarget # {{{ blas callable class BLASCallable(lp.ScalarCallable): - def with_types(self, arg_id_to_dtype, kernel, program_callables_info): + def with_types(self, arg_id_to_dtype, kernel, callables_table): for i in range(0, 2): if i not in arg_id_to_dtype or arg_id_to_dtype[i] is None: # the types provided aren't mature enough to specialize the # callable return ( self.copy(arg_id_to_dtype=arg_id_to_dtype), - program_callables_info) + callables_table) mat_dtype = arg_id_to_dtype[0].numpy_dtype vec_dtype = arg_id_to_dtype[1].numpy_dtype @@ -34,7 +34,7 @@ class BLASCallable(lp.ScalarCallable): from loopy.types import NumpyType return self.copy(name_in_target=name_in_target, arg_id_to_dtype={0: NumpyType(vec_dtype), 1: NumpyType(vec_dtype), - -1: NumpyType(vec_dtype)}), program_callables_info + -1: NumpyType(vec_dtype)}), callables_table def emit_call_insn(self, insn, target, expression_to_code_mapper): assert self.is_ready_for_codegen() diff --git a/loopy/kernel/function_interface.py b/loopy/kernel/function_interface.py index fa7a87fec74bb6910927b56684000e87b111e7f2..3e628f5c9a60f188c0033efa40c5fd762e53ba11 100644 --- a/loopy/kernel/function_interface.py +++ b/loopy/kernel/function_interface.py @@ -532,7 +532,7 @@ class CallableKernel(InKernelCallable): return self.subkernel.name def with_types(self, arg_id_to_dtype, caller_kernel, - program_callables_info): + callables_table): kw_to_pos, pos_to_kw = get_kw_pos_association(self.subkernel) new_args = [] @@ -555,10 +555,10 @@ class CallableKernel(InKernelCallable): # infer the types of the written variables based on the knowledge # of the types of the arguments supplied - specialized_kernel, program_callables_info = ( + specialized_kernel, callables_table = ( infer_unknown_types_for_a_single_kernel( pre_specialized_subkernel, - program_callables_info, + callables_table, expect_completion=True)) new_arg_id_to_dtype = {} @@ -571,9 +571,9 @@ class CallableKernel(InKernelCallable): # Return the kernel call with specialized subkernel and the corresponding # new arg_id_to_dtype return self.copy(subkernel=specialized_kernel, - arg_id_to_dtype=new_arg_id_to_dtype), program_callables_info + arg_id_to_dtype=new_arg_id_to_dtype), callables_table - def with_descrs(self, arg_id_to_descr, program_callables_info): + def with_descrs(self, arg_id_to_descr, callables_table): # tune the subkernel so that we have the matching shapes and # dim_tags @@ -602,15 +602,15 @@ class CallableKernel(InKernelCallable): type(descr)) descriptor_specialized_knl = self.subkernel.copy(args=new_args) from loopy.preprocess import traverse_to_infer_arg_descr - descriptor_specialized_knl, program_callables_info = ( + descriptor_specialized_knl, callables_table = ( traverse_to_infer_arg_descr(descriptor_specialized_knl, - program_callables_info)) + callables_table)) return ( self.copy( subkernel=descriptor_specialized_knl, arg_id_to_descr=arg_id_to_descr), - program_callables_info) + callables_table) def with_packing_for_args(self): from loopy.kernel.data import AddressSpace diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py index 125577c9a0bf7c902686b4b136a7af6ef9201d6d..26856d64fc189191d818bd2ea3eac82c7bfa406e 100644 --- a/loopy/kernel/tools.py +++ b/loopy/kernel/tools.py @@ -1866,7 +1866,7 @@ def find_aliasing_equivalence_classes(kernel): # {{{ callee kernel tools -def get_direct_callee_kernels(kernel, program_callables_info, insn_ids=None,): +def get_direct_callee_kernels(kernel, callables_table, insn_ids=None,): """ Returns an instance of :class:`frozenset` of all the callee kernels called in instructions in the *kernel* whose IDs are given in *insn_ids*. @@ -1892,8 +1892,8 @@ def get_direct_callee_kernels(kernel, program_callables_info, insn_ids=None,): from loopy.kernel.instruction import (CallInstruction, MultiAssignmentBase, CInstruction, _DataObliviousInstruction) if isinstance(insn, CallInstruction): - if insn.expression.function.name in program_callables_info: - in_knl_callable = program_callables_info[ + if insn.expression.function.name in callables_table: + in_knl_callable = callables_table[ insn.expression.function.name] if isinstance(in_knl_callable, CallableKernel): return in_knl_callable.subkernel