config: spell the parallel config tier at the call site (#36250)
This commit is contained in:
@@ -141,7 +141,7 @@ class EagerRunner(BaseRunner):
|
||||
encoder_lens_dtype=(
|
||||
torch.int64 if torch.device(mr.device).type == "cpu" else torch.int32
|
||||
),
|
||||
dp_size=get_parallel().dp_size,
|
||||
dp_size=get_parallel().config.dp_size,
|
||||
)
|
||||
# Eager has no capture step, so warm up here (run-once via mr._kernel_warmed_up).
|
||||
self.warmup()
|
||||
|
||||
Reference in New Issue
Block a user