Skip to content

[bug] Enable RH2D for yuanrong client #125

Description

@tianyi-ge

Background

Now the rh2d client does not specify enable_remote_h2d, so rh2d transport cross node is still via tcp

self._ds_client = datasystem.DsTensorClient(host, port, self.device_id)
self._ds_client.init()

In other words, the benchmark of yuanrong is still running on tcp

Solution

enable it and make it run smoothly. fix potential yuanrong bugs

self._ds_client = datasystem.DsTensorClient(host, port, self.device_id, enable_remote_h2d=True)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions