fix the missing argument in test and typo (#5730)
This PR fixes the issue mentioned in
[PR5722](https://github.com/microsoft/DeepSpeed/pull/5722) that causes
the hangs in the nv-torch-latest-v100 tests.
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
