Skip to content
This repository was archived by the owner on Oct 19, 2024. It is now read-only.

Conversation

haifaksh
Copy link

No description provided.

mesh_id: int = None
):
super().__init__(host_ids, host_info, num_devices_per_host, parent, devices)
self.host_ips = []
Copy link
Collaborator

Choose a reason for hiding this comment

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

This member seems never used. If so, please remove it

self.send_worker_task_ids = {}
self.recv_worker_task_ids = {}

self.task_dones = []
Copy link
Collaborator

Choose a reason for hiding this comment

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

remove unused code. For a series of tasks, we always create a tmp task dones instead of keeping using the same one


def __init__(self,
mesh_group: PhysicalDeviceMeshGroup,
#virtual_mesh_group: VirtualMeshGroup,
Copy link
Collaborator

Choose a reason for hiding this comment

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

remove unused lines. Also applies for some code below

suite.addTest(CreateStateTest("test_shard_parallel"))
suite.addTest(CreateStateTest("test_shard_parallel_grad_acc"))
suite.addTest(CreateStateTest("test_pipeshard_parallel"))
#suite.addTest(CreateStateTest("test_shard_parallel_grad_acc"))
Copy link
Collaborator

Choose a reason for hiding this comment

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

recover the unit test

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants