DEBUG: reproduce flaky test_load_weights_from_remote_instance (#22150)
Co-authored-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
co-authored by
Shangming Cai
parent
df9c831ab8
commit
3a4f4cbc52
@@ -195,9 +195,7 @@ def init_process_dst(
|
||||
remote_instance_weight_loader_send_weights_group_ports=ports,
|
||||
load_format="remote_instance",
|
||||
remote_instance_weight_loader_backend=remote_instance_loader_backend,
|
||||
remote_instance_weight_loader_start_seed_via_transfer_engine=(
|
||||
remote_instance_loader_backend == "transfer_engine"
|
||||
),
|
||||
remote_instance_weight_loader_start_seed_via_transfer_engine=False,
|
||||
)
|
||||
else:
|
||||
host, _, port = DEFAULT_URL_FOR_TEST.rpartition(":")
|
||||
@@ -358,6 +356,7 @@ class TestLoadWeightsFromRemoteInstance(CustomTestCase):
|
||||
assert torch.cuda.device_count() >= 2, "At least 2 GPUs are required"
|
||||
# test_suits : tp, dp, model_name, backend, dst_instance_id
|
||||
if is_in_ci():
|
||||
# FIXME: refactor this test to have less random behavior
|
||||
mode = random.choice(["Engine", "Server"])
|
||||
remote_instance_loader_backend = random.choice(["nccl", "transfer_engine"])
|
||||
test_suits = [
|
||||
|
||||
Reference in New Issue
Block a user