diff --git a/loopy/program.py b/loopy/program.py index 9b71f9d251acd3850fe175c0a471c8fbb98d8e81..4a1225a444eb7a84be3d9e1406d1ae0bf3dca0a6 100644 --- a/loopy/program.py +++ b/loopy/program.py @@ -479,9 +479,10 @@ def _get_callable_ids_for_knl(knl, callables): clbl_id_collector = CallablesIDCollector() return frozenset().union(*( - _get_callable_ids_for_knl(callables[clbl].subkernel, callables) if - isinstance(callables[clbl], CallableKernel) else frozenset([clbl]) - for clbl in clbl_id_collector.map_kernel(knl))) | frozenset([knl.name]) + _get_callable_ids_for_knl(callables[clbl].subkernel, callables) | + frozenset([clbl]) if isinstance(callables[clbl], CallableKernel) else + frozenset([clbl]) + for clbl in clbl_id_collector.map_kernel(knl))) def _get_callable_ids(callables, entrypoints):