fix multiprocess size

jenchen13 · jenchen13 · commit 3f857a32b42f · 2025-10-02T21:17:08.000Z
Signed-off-by: Jennifer Chen &lt;jennifchen@nvidia.com&gt;
diff --git a/tests/gpu/torch/quantization/plugins/test_megatron.py b/tests/gpu/torch/quantization/plugins/test_megatron.py
@@ -199,7 +199,7 @@ def _test_data_tensor_context_parallel_helper(config, rank, size):
 )
 def test_data_tensor_context_parallel(need_8_gpus, config):
     spawn_multiprocess_job(
-        size=4, job=partial(_test_data_tensor_context_parallel_helper, config), backend="nccl"
+        size=8, job=partial(_test_data_tensor_context_parallel_helper, config), backend="nccl"
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -199,7 +199,7 @@ def _test_data_tensor_context_parallel_helper(config, rank, size):`
`199`	`199`	`)`
`200`	`200`	`def test_data_tensor_context_parallel(need_8_gpus, config):`
`201`	`201`	`spawn_multiprocess_job(`
`202`		`- size=4, job=partial(_test_data_tensor_context_parallel_helper, config), backend="nccl"`
	`202`	`+ size=8, job=partial(_test_data_tensor_context_parallel_helper, config), backend="nccl"`
`203`	`203`	`)`
`204`	`204`
`205`	`205`