implements concatenate_calls

Author: kaushikcfd

pytato/__init__.py

@@ -104,6 +104,7 @@ def set_debug_enabled(flag: bool) -> None:
                                 execute_distributed_partition,
                                 )
 from pytato.tracing import trace_call
+from pytato.transform.calls import concatenate_calls
 
 from pytato.partition import generate_code_for_partition
 
@@ -152,6 +153,8 @@ def set_debug_enabled(flag: bool) -> None:
 
         "trace_call",
 
+        "concatenate_calls",
+
         "make_distributed_recv", "make_distributed_send", "DistributedRecv",
         "DistributedSend", "staple_distributed_send", "DistributedSendRefHolder",