Implement pt.concatenate_calls

Author: kaushikcfd

pytato/__init__.py

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