DeepSpeed
ae075141 - Remove stale step() docstring from DeepSpeedCPUAdam (#8011)

Commit
89 days ago
Remove stale step() docstring from DeepSpeedCPUAdam (#8011) ## What Removes an obsolete paragraph from the `DeepSpeedCPUAdam` constructor docstring. ## Why The docstring described a `step()` option that "updates optimizer states and copies the parameters back to GPU at the same time" — the old `adam_update_copy` kernel, invoked via `step(fp16_param_groups=...)`. That fused-copy path no longer exists in the codebase: - `csrc/adam/cpu_adam.cpp` binds only `adam_update` (no `adam_update_copy`). - `DeepSpeedCPUAdam.step()` / `step_subgroup()` take no `fp16_param_groups` argument and only call `adam_update`. So the "two options" text is stale and misleading to anyone reading the API. Docstring-only change; no functional impact. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Signed-off-by: Lucas Pirola <lucas@pirola.eu> Co-authored-by: Claude Opus 4.7 <noreply@anthropic.com>
Author
Parents
Loading