add deprecated deepspeed flag for legacy code (#104)

5897091e · Jeff Rasley · GitHub · 853a3a92 · 5897091e
Unverified Commit 5897091e authored Feb 24, 2020 by Jeff Rasley Committed by GitHub Feb 24, 2020
Show whitespace changes
Inline Side-by-side

Showing with 14 additions and 4 deletions

deepspeed/__init__.py deepspeed/__init__.py +14 -4

No files found.
--- a/deepspeed/__init__.py
+++ b/deepspeed/__init__.py
@@ -118,16 +118,26 @@ def _add_core_arguments(parser):
    """
    group = parser.add_argument_group('DeepSpeed', 'DeepSpeed configurations')

-    group.add_argument('--deepspeed',
+    group.add_argument(
+        '--deepspeed',
        default=False,
        action='store_true',
-                       help='Enable DeepSpeed')
+        help=
+        'Enable DeepSpeed (helper flag for user code, no impact on DeepSpeed backend)')

    group.add_argument('--deepspeed_config',
                       default=None,
                       type=str,
                       help='DeepSpeed json configuration file.')

+    group.add_argument(
+        '--deepscale',
+        default=False,
+        action='store_true',
+        help=
+        'Deprecated enable DeepSpeed (helper flag for user code, no impact on DeepSpeed backend)'
+    )
+
    group.add_argument('--deepscale_config',
                       default=None,
                       type=str,