Skip to content

Conversation

Wanglongzhi2001
Copy link
Collaborator

@Wanglongzhi2001 Wanglongzhi2001 commented Aug 21, 2025

Support streaming transfer data using ZMQ

Already verified cases:

  • mixed EP in single node (text mode without mtp)
  • TP in single node (text mode without mtp)

TODO:

  • accelerate the serialize and deserialize process
  • adapt and verify logprob
  • adapt and verify speculate decoding

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.


Wanglongzhi2001 seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

Copy link

paddle-bot bot commented Aug 21, 2025

Thanks for your contribution!

self.zmq_server = ZmqClient(name=f"get_save_output_rank{self.cfg.local_device_ids[0]}", mode=zmq.PULL)
self.zmq_server.start_server()
self.zmq_server.create_router()
time.sleep(3)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里sleep的目的是为了等待初始化完成还是建联呢 是否有更优雅的方式



def post_process_normal(
cls,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里是不是还有其他硬件的调用也需要去改。以及这里传递zmq_client不行吗 还要把cls传过来

@Wanglongzhi2001 Wanglongzhi2001 force-pushed the new_stream_transfer branch 6 times, most recently from c850e37 to 361c4eb Compare August 31, 2025 07:19
Copy link
Collaborator

@RichardWooSJTU RichardWooSJTU left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@RichardWooSJTU RichardWooSJTU merged commit e0c9a6c into PaddlePaddle:develop Sep 2, 2025
34 of 39 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants