fixed incorrect kwarg name in TP overlap test

Signed-off-by: Alp Dener <[email protected]>

tests/pytorch/distributed/run_layer_with_overlap.py

@@ -266,7 +266,7 @@ def dist_print(msg, src=None, end="\n", debug=False, error=False):
         use_fp8=opts.fp8,
         dtype=torch.bfloat16,
         bootstrap_backend=opts.bootstrap_backend,
-        ub_cfg=_get_ub_cfg(opts.layer_type),
+        ub_cfgs=_get_ub_cfg(opts.layer_type),
     )
 
     # Initialize the Transformer Engine layer with overlap