Currently, the FX graph tracing (such as the one used in aot_module) seems not supporting collective functions such as allgather. We may face some errors like ...